2020年4月27日
【转载】PyTorch网络总结
摘要： 1、各网络比较 https://www.zybuluo.com/rianusr/note/1419006 2、卷积神经网络之AlexNet 阅读全文
posted @ 2020-04-27 15:53 夕西行 阅读(160) 评论(0) 推荐(0)
PyTorch保存、加载模型，PyTorch中已封装的网络模型
摘要： state_dict()函数可以返回所有的状态数据。load_state_dict()函数可以加载这些状态数据。 推荐使用： #保存 t.save(net.state_dict(),"net.pth") #加载 net2=Net() net2.load_state_dict(t.load("net. 阅读全文
posted @ 2020-04-27 14:01 夕西行 阅读(2110) 评论(0) 推荐(0)