timm.create_model()下载预训练的权重会网络报错
先打印网络权重的地址
pretrained_cfg = timm.models.create_model(cfg.backbone).default_cfg
print(pretrained_cfg)
#再用windows电脑科学上网,下载这个权重
https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b3_ns-9d44bf68.pth
#再写一个路径指向这个权重
pretrained_cfg_overlay=dict(file='./xxx/tf_efficientnet_b3_ns-9d44bf68.pth')
import timm
in_chans = cfg.n_channels
encoder = timm.create_model(
cfg.backbone,
pretrained=cfg.pretrained,
features_only=True,
in_chans=in_chans,
pretrained_cfg_overlay=dict(file=
'/xxx/tf_efficientnet_b3_ns-9d44bf68.pth')
)