1. 首页
  2. 编程语言
  3. Python
  4. 解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题

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

上传者: 2022-02-19 22:14:59上传 PDF文件 63.69 KB 热度 11次

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

下载地址
用户评论