
如何将pytorch多GPU训练的模型在单GPU环境中成功加载?
本文共计698个文字,预计阅读时间需要3分钟。背景:在公用实验室使用多卡训练模型,获取权值文件后保存,然后回到实验室,没有多卡的环境,使用单卡训练,加载模型时出错,因为单卡机器上没有使用DataParallel来加载模型。结果:出现错误提示
共收录篇相关文章

本文共计698个文字,预计阅读时间需要3分钟。背景:在公用实验室使用多卡训练模型,获取权值文件后保存,然后回到实验室,没有多卡的环境,使用单卡训练,加载模型时出错,因为单卡机器上没有使用DataParallel来加载模型。结果:出现错误提示