假设我们只保存了模型的参数(model.state_dict())到文件名为modelparameters.pth, model = Net()

1. cpu -> cpu或者gpu -> gpu:

checkpoint = torch.load('modelparameters.pth')

model.load_state_dict(checkpoint)

2. cpu -> gpu 1

torch.load('modelparameters.pth', map_location=lambda storage, loc: storage.cuda(1))

3. gpu 1 -> gpu 0

torch.load('modelparameters.pth', map_location={'cuda:1':'cuda:0'})

4. gpu -> cpu

torch.load('modelparameters.pth', map_location=lambda storage, loc: storage)




 
												


											
pytorch将cpu训练好的模型参数load到gpu上，或者gpu->cpu上的更多相关文章
	

    
								
  1. pytorch 加载训练好的模型做inference
		
    前提: 模型参数和结构是分别保存的 1. 构建模型(# load model graph) model = MODEL() 2.加载模型参数(# load model state_dict) mode ...
    
		
    2. 
						
  2. gluoncv 用已经训练好的模型参数，检测物体
		
    当然这个模型参数,最好用自己的,否则不够精确,我自己的还没训练完. from matplotlib import pyplot as plt import gluoncv from gluoncv i ...
    
		
    3. 
						
  3. tflearn kears GAN官方demo代码——本质上GAN是先训练判别模型让你能够识别噪声，然后生成模型基于噪声生成数据，目标是让判别模型出错。GAN的过程就是训练这个生成模型参数！！！
		
    GAN:通过 将 样本 特征 化 以后, 告诉 模型 哪些 样本 是 黑 哪些 是 白, 模型 通过 训练 后, 理解 了 黑白 样本 的 区别, 再输入 测试 样本 时, 模型 就可以 根据 以往  ...
    
		
    4. 
						
  4. libsvm 训练后的模型参数讲解（转）
		
    主要就是讲解利用libsvm-mat工具箱建立分类(回归模型)后,得到的模型model里面参数的意义都是神马?以及如果通过model得到相应模型的表达式,这里主要以分类问题为例子.测试数据使用的是li ...
    
		
    5. 
						
  5. [转]libsvm 训练后的模型参数讲解
		
    http://blog.sina.com.cn/s/blog_6646924501018fqc.html 主要就是讲解利用libsvm-mat工具箱建立分类(回归模型)后,得到的模型model里面参数 ...
    
		
    6. 
						
  6. Pytorch使用分布式训练，单机多卡
		
    pytorch的并行分为模型并行.数据并行 左侧模型并行:是网络太大,一张卡存不了,那么拆分,然后进行模型并行训练. 右侧数据并行:多个显卡同时采用数据训练网络的副本. 一.模型并行 二.数据并行 数 ...
    
		
    7. 
						
  7. pytorch_模型参数-保存，加载，打印
		
    1.保存模型参数(gen-我自己的模型名字) torch.save(self.gen.state_dict(), os.path.join(self.gen_save_path, 'gen_%d.pt ...
    
		
    8. 
						
  8. [深度学习] Pytorch（三）—— 多/单GPU、CPU，训练保存、加载模型参数问题
		
    [深度学习] Pytorch(三)-- 多/单GPU.CPU,训练保存.加载预测模型问题 上一篇实践学习中,遇到了在多/单个GPU.GPU与CPU的不同环境下训练保存.加载使用使用模型的问题,如果保存 ...
    
		
    9. 
						
  9. pytorch入门2.2构建回归模型初体验（开始训练）
		
    pytorch入门2.x构建回归模型系列: pytorch入门2.0构建回归模型初体验(数据生成) pytorch入门2.1构建回归模型初体验(模型构建) pytorch入门2.2构建回归模型初体验( ...
    
		
    10. 
		

	


随机推荐
	

    
									
  1. Spring注解和JDK注解
			
    1.添加xsd约束 xmlns:context="http://www.springframework.org/schema/context" http://www.springf ...
    
			
    2. 
						
  2. drupal6提示 Compilation failed: disallowed Unicode code point (>= 0xd800 && <= 0xdfff) at offset 9 on line 615
			
    解决办法:将sites\all\modules\ctools\includes\cleanstring.inc文件中的61行改成62行这样子即可,如下图
    
			
    3. 
						
  3. user-select详解
			
    语法: user-select:none | text | all | element 默认值:text 适用于:除替换元素外的所有元素 继承性:无 动画性:否 计算值:指定值 取值: none: 文 ...
    
			
    4. 
						
  4. SQLServer怎样把本地数据导入到远程服务器上(转载)
			
    平常用到mssql时间比较少,总是过一段时间就忘记应该怎么操作了.当要做mssq把本地数据导入到远程服务器的时候,就去网上搜索很久都没有图解的,所以今天自己收集一下免得下次又到处去找.希望对自己,同时 ...
    
			
    5. 
						
  5. 洛谷 P1363 幻想迷宫
			
    题目描述 背景 Background (喵星人LHX和WD同心协力击退了汪星人的入侵,不幸的是,汪星人撤退之前给它们制造了一片幻象迷宫.) WD:呜呜,肿么办啊…… LHX:momo...我们一定能走 ...
    
			
    6. 
						
  6. openstack 存储节点按照报错Device /dev/sdb not found (or ignored by filtering).
			
    root@dell-PowerEdge-T30:~# pvcreate /dev/sdb  Device /dev/sdb not found (or ignored by filtering).首页 ...
    
			
    7. 
						
  7. python实现微信打飞机游戏(by crossin)
			
    # -*- coding: utf-8 -*- import pygame from sys import exit import random pygame.init() screen = pyga ...
    
			
    8. 
						
  8. Android计算器布局
			
    Android(安桌)计算器布局实现         ——解决整个屏幕方案 引言:     学完了android布局的几种方式,做了一个android计算器. 我在网上搜索了这方面的资料,发现了布局都 ...
    
			
    9. 
						
  9. java 使用htmlunit模拟登录爬取新浪微博页面
			
    mport java.io.IOException;import java.net.MalformedURLException;import com.gargoylesoftware.htmlunit ...
    
			
    10. 
						
  10. HDU 4341 Gold miner （分组背包）
			
    先把线按照距离原点的距离排序,然后用叉积把在同一条直线上的点放在一起, 把在同一条线上的点中的前i个点当成一个点就转化成了分组背包. 写if(kas++) putchar('\n') 居然PE了,PE ...
    
			
    11.