def set_input(self, input):
    self.real_A = input['A'].to(self.device)
    self.image_paths = input['A_paths']
def set_input(self, input):
    self.real_A = input['A'].to(self.device)
    self.image_paths = input['A_paths']