torch.load
加载的是训练好的模型
例如
torch.load("ad.pth")
而load_state_dict是net的一个方法
是将torch.load加载出来的数据加载到net中
torch.load
加载的是训练好的模型
例如
torch.load("ad.pth")
而load_state_dict是net的一个方法
是将torch.load加载出来的数据加载到net中
上一篇:一名大神的github
下一篇:美丽的开关
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M