今天跑一份代码,出现这样的报错信息:

AttributeError: 'VisionTransformer' object has no attribute "'pre_logits'"

VisionTransformer这个类是timm库里面直接导入的,查看github上源代码发现VisionTransformer里确实有pre_logits这个属性。那么解决方案就是把timm更新到最新版本:

pip install --upgrade timm

使用pytorch时遇到类似的问题也可以尝试更新package

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐