adv-efficientnet-b3-cdd7c0f4.pth
PyTorch EfficientNet
在线加载模型地址

确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程

在代码中实现加载预训练模型地址

调用模型的实现方法

from efficientnet_pytorch import EfficientNet

model = EfficientNet.from_pretrained("efficientnet-b3", advprop=True)

if advprop: # for models using advprop pretrained weights normalize = transforms.Lambda(lambda img: img * 2.0 - 1.0)else: normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])

立刻学会使用预训练模型!

查看不同深度学习框架下使用预训练模型的样例代码