在公司用多卡训练模型,得到权值文件后保存,然后回到实验室,没有多卡的环境,用单卡训练,加载模型时出错,因为单卡机器上,没有使用DataParallel来加载模型,所以会出现加载错误。测试环境:Python3.6 + Pytorch0.4在pytorch中,使用多GPU训练网络需要用到 【而使用单GPU训练网络:

解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题

解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题