今天跑一份代码,出现这样的报错信息:
AttributeError: 'VisionTransformer' object has no attribute "'pre_logits'"
VisionTransformer这个类是timm库里面直接导入的,查看github上源代码发现VisionTransformer里确实有pre_logits这个属性。那么解决方案就是把timm更新到最新版本:
pip install --upgrade timm
使用pytorch时遇到类似的问题也可以尝试更新package