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'

[Xuyang Wang]

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

[Xuyang Wang]

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

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

[pt]

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

[Laniakea928]

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

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