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

problems about model saving #120

Open HuaZheLei opened 4 years ago

HuaZheLei commented 4 years ago

Hi, thanks for your excellent work. You use 'DataParallel' to support multi-gpu training. The standard way for saving models with 'DataParallel' is to use torch.save(model.module.state_dict(), PATH). However, in your code, you just use torch.save({ 'discriminator': self.discriminator.state_dict() }, self.dis_weights_path).

anshen666 commented 4 years ago

你好，我最近也在跑这个代码。可以加你交流一下吗？我的微信：loveanshen 我的QQ：519838354 我的邮箱：519838354@qq.com 非常期待你百忙中的回复