AttributeError: 'resnet_swap_2loss_add' object has no attribute 'module'

[MercuialC]

感谢您优秀的工作！
我使用了您的环境，但是#172 if isinstance(m.module, Classifier) or 'lrx' in key:...此处报错
AttributeError: 'resnet_swap_2loss_add' object has no attribute 'module'

[xuyangwang0825]

相同的问题，想问下你解决了吗？

[xuyangwang0825]

相同的问题，想问下你解决了吗？

直接把这个判断注释掉就可以了

[peggysoul]

同样的问题+1，请问你解决了吗

[Laniakea928]

相同的问题，想问下你解决了吗？

直接把这个判断注释掉就可以了
请问单注释掉这个判断能顺利运行吗？

[zlsbksdxl]

If conditions permit, you can use two GPUs for training, which will prevent such errors.