代码小错误

[yangyujie0412]

在运行代码的过程中发现了两个小错误 add_weight_decay ——> param_groups_weight_decay module ——> _modules

[jihaonew]

Thanks for the pointer. The error may come from the different versions of PyTorch and Timm. The code is tested under torch==1.11.0 timm==0.5.4. I will add a requirements.txt for a better implementation.

感谢指出，这个问题可能来自于不同的pytorch和timm的版本。我们的代码是在 torch==1.11.0 timm==0.5.4的情况下测试的。后续我们会加一个requirements.txt。