“Discriminator.b16.conv1.weight”出现异常

Ldhlwh / DFMGAN

[AAAI 2023] Official implementation of "Few-Shot Defect Image Generation via Defect-Aware Feature Manipulation".

https://arxiv.org/abs/2303.02389

Other

83 stars 11 forks source link

Closed 153201121yf closed 5 months ago

153201121yf commented 5 months ago

tick训练到50的时候突然报错tensor和other的值不相等，是什么原因呢，该如何解决

Ldhlwh commented 5 months ago

我之前没遇到过类似情况，看起来是stylegan部分多卡ddp出的问题，可以尝试重新训练，如果问题持续出现可以尝试用单卡或者到原stylegan的repo找找解决方案