def save_image(self, input, result_dir, filename, logger):
img = world.sample2img(input.to("cpu"), self.height, self.width)
image_name = os.path.join(result_dir, filename)
def save_image(self, input, result_dir, filename, logger):
img = world.sample2img(input.to("cpu"), self.height, self.width)
image_name = os.path.join(result_dir, filename)