resnest50-528c19ca.pth
PyTorch ResNeSt
在线加载模型地址

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

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

调用模型的实现方法

from flyai.utils import remote_helper

path = remote_helper.get_remote_data('https://www.flyai.com/m/resnest50-528c19ca.pth')

from resnest.torch import resnest50 

model = resnest50(pretrained=False)

state = torch.load(path) 

model.load_state_dict(state, strict=True) 

feature = model.fc.in_features 

# num_classes 为类别数

model.fc = nn.Linear(in_features=feature, out_features=num_classes, bias=True)