地址链接:https://www.flyai.com/m/hrnetv2_w18_imagenet_pretrained.pth
查看来源:https://github.com/HRNet/HRNet-Image-Classification
确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程
在代码中实现加载预训练模型地址
import torch
from flyai.utils import remote_helper
model = HRnet()
path = remote_helper.get_remote_date('https://www.flyai.com/m/hrnetv2_w18_imagenet_pretrained.pth')
dict = torch.load(path)
model.load_state_dict(dict)