为什么我会收到错误: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 标记错误。