search

knazeri / edge-connect

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212
http://openaccess.thecvf.com/content_ICCVW_2019/html/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.html
Other
2.52k stars 532 forks source link

only got epoch 1 #85

Closed JenicChen closed 5 years ago

JenicChen commented 5 years ago

Hi Knazeri, I use celeba for train which has 202,599 pics, and my mask files were augmented to 202,599 too(from 12,000).I'm sure that I made correct flist,but the result is that I just got one epoch,which ends right after. image

knazeri commented 5 years ago

@JenicChen Are you using the pre-trained weights? I see maximum iterations are set to 20,000, so if you are using the pre-trained weights which are trained up to 2,000,000 then training stops! Try to set maximum iterations to a larger value like MAX_ITERS: 2e7

JenicChen commented 5 years ago

@knazeri it does work!Thank you so much!