使用Pytorch在多GPU下保存和加载训练模型参数遇到的问题
原文转载至https://www.cnblogs.com/jinjunweina/p/12671833.html使用Pytorch在多GPU下保存和加载训练模型参数遇到的问题最近使用Pytorch在学习一个深度学习项目,在模型保存和加载过程中遇到了问题,最终通过在网卡查找资料得已解决,故以此记之,以备忘却。首先,是在使用多GPU进行模型训练的过程中,在保存模型参数时,应该使用类似如下代码进行保存:torch.save({ 'epoch': epoch,
转载
2020-11-09 09:29:53 ·
1904 阅读 ·
0 评论