neural_style.py 文件源码

def fit(self, content_image, style_image, epoch_num, callback=None):
        device_id = None
        if self.device_id >= 0:
            device_id = self.device_id
        with cuda.get_device_from_id(device_id):
            return self.__fit(content_image, style_image, epoch_num, callback)