Mistakes in the forward functions in models.py

[hedjm]

Dear sir, Thank you for your great work.

I think there are some errors in models.py code (in the forward functions in both models). edges_masked = (edges * (1 - masks)) images_masked = (images * (1 - masks)) + masks I used the debugger to make sure the values are correct and also I wrote the images in disk. The code should be: edges_masked = edges * masks images_masked = images * masks I may be wrong, but please check it.

Thank you.

[anshen666]

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