torch.load、torch.save、torch.optim.Adam的用法
一、torch.save()
1.只保存model的权重
2.保存多项内容
二、torch.load()
三、torch.optim.Adam
1.optim.Adam()参数说明
optim.Adam()(params, lr=0.001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0)
参考(147条消息) pytorch 中 torch.optim.Adam 方法的使用和参数的解释_Ibelievesunshine的博客-CSDN博客_t
原创
2021-10-28 17:57:03 ·
9088 阅读 ·
0 评论