torch.save,为cpu设置种子,model.eval()与torch.no_grad(),model.forward()与model()

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值