search

avalonstrel / GatedConvolution

An reimplement version of inpainting model in Free-Form Image Inpainting with Gated Convolution
Other
115 stars 27 forks source link

About inpaint.yml #5

Open JinMengKang opened 6 years ago

JinMengKang commented 6 years ago

Dear author, in inpaint.yml, I am confused about two parameters TRAIN_SPE: 10000 MAX_ITERS: 1000000 When I set these two parameters： TRAIN_SPE: 3000 MAX_ITERS: 1000 Training stopped after four iterations, and the result was also very bad. What do these two parameters mean and how to determine the number of training iterations? Thank you for your patience.

JinMengKang commented 6 years ago

And I only have 3000 pictures.How should I set the parameters to ensure a sufficient number of trainings

avalonstrel commented 6 years ago

Actually, these parameters are used in generative_inpainting(https://github.com/JiahuiYu/generative_inpainting) first, and I found it in trainer.py, it is used for logging and should be smaller than MAX_ITERS in my view, you may try 500, 1000 for training.