search

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
22.78k stars 6.29k forks source link

there is no `latest_net_G.pth` in the directory where checkpoints are being saved #1671

Open Rjaat opened 3 weeks ago

Rjaat commented 3 weeks ago

I have trained model on my own dataset for 200 epochs. latest weights are saved as latest_net_G_A.pth, latest_net_G_B.pth, latest_net_D_A.pth, and latest_net_D_B.pth

After completing this, I tried to run test.py for the trained model on unseen data. but it throws error as FileNotFoundError: [Errno 2] No such file or directory: './checkpoints/DNI_GAN/latest_net_G.pth'

My Quesries:

  1. I dont know, why it's not being saved?
  2. I to test images from domain A to domain B or vice versa???
master2021-bit commented 2 weeks ago

There maybe a latest_net_G_A.pth and latest_net_G_B.pth at somewhere, you may find it, then you can build a new directory './checkpoints/DNI_GAN and take the .pth to the new directory

kovalepskaia commented 2 weeks ago

you read section Train in this https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/blob/master/CycleGAN.ipynb