翻墙去huggingface的timm中搜索对应预训练模型文件。https://huggingface.co/timm
比如下载maxvit_tiny_rw_224(rw是对应pytorch使用的版本,tf则是tensorflow)。
下载pytorch_model.bin文件即可。
然后将该文件放到XXX/.cache/torch/hub/checkpoints目录下。
(改名版)
在使用时,第一个参数为文件名,将pretrained设为false(这样就不会再从网络上下载模型文件,直接使用本地checkpoints中的)。
import timm
#设置模型
model_ft=timm.create_model("maxvit_tiny_rw_224",pretrained=False)