Pytorch-cyclegan-and-pix2pix: How to change the max epoch when training?

[zhouyingji]

The default epoch is 200, but how to change it? I can't find which parameter to adjust it.
IF I want to training 400 epoch?
Thank you

[smitheric95]

From the train.py --help output:

  --n_epochs N_EPOCHS   number of epochs with the initial learning rate
                        (default: 100)
  --n_epochs_decay N_EPOCHS_DECAY
                        number of epochs to linearly decay learning rate to
                        zero (default: 100)

So in your case, python train.py --n_epochs 300 --n_epochs_decay 100

[zhouyingji]

From the train.py --help output:
--n_epochs N_EPOCHS number of epochs with the initial learning rate
(default: 100)
--n_epochs_decay N_EPOCHS_DECAY
number of epochs to linearly decay learning rate to
zero (default: 100)

So in your case, python train.py --n_epochs 300 --n_epochs_decay 100

Get it! Thank you!