Thank you, I’ll try that.
I save the model as checkpoints during training, and all predictions, also from the first saved model, look similar to the last saved model. I think maybe there is problem with saving the model. But I dont think I’m doing anything wrong here:
def save_state(self, path_save='/Model'):
        t = time.localtime()
        torch.save(self.G_AB.state_dict(), self.run_dir+'/'+path_save+'/'+time.strftime("%H-%M-%S_G_AB.pt", t))
        torch.save(self.G_BA.state_dict(), self.run_dir+'/'+path_save+'/'+time.strftime("%H-%M-%S_G_BA.pt", t))

Thanks for your help .