2019年11月5日

pytorch加载模型
摘要： 1.加载全部模型： net.load_state_dict(torch.load(net_para_pth)) 2.加载部分模型 net_para_pth = './result/5826.pth'pretrained_dict = torch.load(net_para_pth)model_dic 阅读全文
posted @ 2019-11-05 10:24 江南烟雨尘 阅读(4502) 评论(4) 推荐(0) 编辑

导航