How to choose which training results to display ?

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

Other

22.8k stars 6.29k forks source link

How to choose which training results to display ? #1421

Open AnnaBoim opened 2 years ago

AnnaBoim commented 2 years ago

Hi, I'm training Pix2Pix on my own dataset and i want to compare the training results after each epoch. For this i need the same image to be displayed in the web folder after each epoch. Currently, after each epoch a different image is displayed (i don't know if they're chosen randomly or by the best result or something else?) Where can i control which images will be saved for each epoch and their amount (if i want more then one)?

Thanks

junyanz commented 2 years ago

Currently, images are chosen randomly based on the data loader. To achieve your goal, you can use the test script with different epoch numbers. Or you can try to modify the visualization code here.