import timm
from timm.models.efficientnet import _cfg
config = _cfg(url='', file='tf_efficientnetv2_s_21ft1k-d7dafa41.pth') #file为本地文件路径
model = timm.create_model(
"tf_efficientnetv2_s_in21ft1k",
pretrained=True,
features_only=True,
pretrained_cfg=config
)
timm创建预训练模型
最新推荐文章于 2023-12-04 21:34:14 发布