最新的timm版本(0.9.x)默认使用huggingface hub作为权重,优先于torch hub缓存,许多权重已经为模型更改重新映射,所以最好通过HF hub下载。Kaggle真的应该支持通过HF hub,或者至少正确地缓存它,但他们似乎没有兴趣让事情变得更简单,所以手动缓存数据集中的权重的疯狂继续…
要覆盖预训练的位置,您可以从HF hub手动下载权重文件并尝试以下操作。
timm.create_model(
'swinv2_large_window12to16_192to256',
pretrained=True,
pretrained_cfg_overlay=dict(file='path/to/checkpoint'),
)
https://github.com/huggingface/pytorch-image-models/discussions/1826#discussioncomment-5996455