RuntimeError while train model from running from pretrained model places2

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.5k stars 530 forks source link

RuntimeError while train model from running from pretrained model places2 #132

Open goldservice2017 opened 4 years ago

goldservice2017 commented 4 years ago

Hi @knazeri I have checked your pretrained model in testing mode, and working to train it with places2 dataset. I've downloaded latest model from your README. When I started the training, it causing runtime error and I couldn't figure out how to fix this issue. TinyTake24-11-2019-12-22-49_637101229724951508

Please give me solution to fix this issue. Looking forward to hearing from you instantly. Thank you

goldservice2017 commented 4 years ago

Anyone can help me to fix this issue?