search

NVIDIA / pix2pixHD

Synthesizing and manipulating 2048x1024 images with conditional GANs
https://tcwang0509.github.io/pix2pixHD/
Other
6.62k stars 1.39k forks source link

How to run the training cycle for more than 200 Epochs? #307

Closed AlexMackay94 closed 2 years ago

AlexMackay94 commented 2 years ago

Hello!

I cannot find the code to run the training cycle for more than 200 epochs. Is this in the base options, training options, or elsewhere?

Thank you.

AlexMackay94 commented 2 years ago

It was pointed out that the code for this was within train_options.py.

self.parser.add_argument('--niter', type=int, default=100, help='# of iter at starting learning rate')

I added --niter 200 to my training code and the process ran for 400 rather than 200 epochs.