pytorch-CycleGAN-and-pix2pix icon indicating copy to clipboard operation
pytorch-CycleGAN-and-pix2pix copied to clipboard

how much epoch I need to get the realistic results?

Open elsobhano opened this issue 2 years ago • 5 comments

Hi everybody

I trained my network for 30 epochs. Is it normal that I don't have good results?

elsobhano avatar Mar 07 '22 15:03 elsobhano

It could be the case. If possible, just try running for more epochs and compare results. Also, you can use checkpoints to verify that the model is still improving.

tarmopungas avatar Mar 08 '22 09:03 tarmopungas

It depends on your data. Intuitively, however, >100 epoch is more likely to give promising result. I've tried pix2pix on my own dataset and it starts to generate good result after 100 epoch. Also you could try learning rate decay.

zihaog0724 avatar Mar 08 '22 09:03 zihaog0724

It depends on many factors: the number of images in your dataset, the complexity of your input/output domains, the task itself, the image resolution, etc. 30 epochs might not be enough unless you have lots of images in your dataset.

junyanz avatar Mar 08 '22 21:03 junyanz

Hi all my training take long time to get the result 1 hour per epoch, is it normal ? note: the input image size 512x512 can i stop the training in any step i want??

Sarmadfismael avatar Mar 14 '22 20:03 Sarmadfismael

@Sarmadfismael try reducing the image size (using crop or other options listed in the Training/test tips section). Also, it will depend on the memory available on your setup. As far as I understood, there is an option to resume training if needed. So, yes you can stop and resume it. Search for the --continue_train flag in the Training/Test tips page.

ankur-chr avatar Jul 19 '22 13:07 ankur-chr