为什么我会收到错误:ZeroDivisionError:0.0 从 Timm 加载 deit 模型时无法提升为负功率错误

社会演员多 pytorch 257

原文标题Why do I get the error: ZeroDivisionError: 0.0 cannot be raised to a negative power error when loading deit model from Timm

我正在尝试对来自timm 的 DeiT 小型变体进行推断。

from timm.models import create_model
model = create_model('deit_small_patch16_224', pretrained=True)

但我得到了错误:

self.scale = head_dim ** -0.5
ZeroDivisionError: 0.0 cannot be raised to a negative power.

然而,用model = create_model('deit_tiny_patch16_224', pretrained=True)创建一个不同的模型,我可以成功地进行推理,而且效果很好。我了解此错误是由于零除以非零值或被提升到非零次方的结果。但我不太明白为什么这个模型甚至在加载任何数据之前就从 timm 标记错误。

原文链接:https://stackoverflow.com//questions/71910237/why-do-i-get-the-error-zerodivisionerror-0-0-cannot-be-raised-to-a-negative-po

回复

我来回复
  • Craving_gold的头像
    Craving_gold 评论

    这是我的 timm 版本的错误。升级timm:pip install --upgrade timm解决问题。

    2年前 0条评论