摘要: 在训练前写model.train(),进入训练模式。 在预测前写model.eval(),进入预测模式。 参考https://blog.csdn.net/weixin_43593330/article/details/107547202 阅读全文
posted @ 2020-11-29 19:36 火锅先生 阅读(1909) 评论(0) 推荐(0) 编辑
摘要: 一般保存和加载模型的后缀名都是.ph或者.pth 官方文档 保存与加载整个模型(不推荐) 该方法会保存整个模型的网络架构以及当前训练的参数(w、b等) 保存模型 torch.save(model, PATH) 加载模型 model = torch.load(PATH) 举例: torch.save( 阅读全文
posted @ 2020-11-29 19:22 火锅先生 阅读(253) 评论(0) 推荐(0) 编辑