MEALV2_ResNet50_224.pth
PyTorch CNN
在线加载模型地址

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

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

调用模型的实现方法

from torchvision.models import resnet50 

from flyai.utils import remote_helper 

from torch import nn 

 path = remote_helper.get_remote_data("https://www.flyai.com/m/MEALV2_ResNet50_224.pth") 

model = resnet50(pretrained=False) 

model.load_state_dict(torch.load(path)) 

 model.fc = nn.Linear(2048, 10)