search

researchmm / AOT-GAN-for-Inpainting

[TVCG'2023] AOT-GAN for High-Resolution Image Inpainting (codebase for image inpainting)
https://arxiv.org/abs/2104.01431
Apache License 2.0
403 stars 64 forks source link

training failed #18

Open jackleibest opened 1 month ago

jackleibest commented 1 month ago

微信截图_20240529095756

got the error on training

ads2d commented 1 month ago

I also have this problem, have you solved it?

s8088899 commented 1 month ago

the same problem，that’s why？？？

s8088899 commented 1 month ago

I solved the problem.Just remove the .module from lines 79 and 82 of the trainer.py file,but the results need to be validated : (