timm.create_model() connect error

最新推荐文章于 2025-03-24 12:23:02 发布

原创最新推荐文章于 2025-03-24 12:23:02 发布 · 862 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#python

文章讲述了如何在Windows系统中使用timm库下载并加载预训练的tf_efficientnet_b3_ns模型权重，涉及科学上网和权重路径配置。

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')
        )