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