torch.save
转载1:https://zhuanlan.zhihu.com/p/67053004?from_voters_page=true
首先讲讲保存模型或权重参数的后缀格式,权重参数和模型参数的后缀格式一样,pytorch中最常见的模型保存使用 .pt 或者是 .pth 作为模型文件扩展名。还有其他的保存数据的格式为.t7或者.pkl格式。t7文件是沿用torch7中读取模型权重的方式,而pth文件是python中存储文件的常用格式,而在keras中则是使用.h5文件 。
转载2:https://blog.csdn.net/qq_16792139/article/details/108658953
state = {‘net’:model.state_dict(), ‘optimizer’:optimizer.state_dict(), ‘epoch’:epoch}
为cpu设置种子
转载1:https://blog.csdn.net/weixin_43002433/article/details/104706950
调用了两次torch.rand这个API (随机生成[0, 1]的数),可以发现输出的是两个不同的随机数,不是说已经固定种子了吗?怎么还会生成不同的随机数?实际上,准确来说,手动设置种子之后导致的结果应该是:在每次重新运行上面的程序时,得到的打印结果都是上面两个数。
model.eval()与torch.no_grad()
转载1:https://blog.csdn.net/qq_37555071/article/details/107536530
转载2:https://blog.csdn.net/songyu0120/article/details/103884586
model.forward()与model()
转载1:https://blog.csdn.net/HJC256ZY/article/details/106462417