假设我们只保存了模型的参数(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. linux下mysql中文乱码
			
    登录mysql执行mysql> show variables like 'character%';发现编码有些不是utf-8 修改/etc/mysql/my.cnf,网上说的是/etc/my.c ...
    
			
    2. 
						
  2. MySQL出现GROUP BY clause错误解决办法
			
    #1)当mysql数据库出现如下错误:#which is not functionally dependent on columns in GROUP BY clause; this is incom ...
    
			
    3. 
						
  3. iOS5 and iOS6都只支持横屏的方法
			
    If your app uses a UINavigationController, then you should subclass it and set the class in IB. You  ...
    
			
    4. 
						
  4. 创建一个文件夹用于写入UTF-8编码的文件
			
    实现效果: 知识运用: File类的CreateText方法 StreamWriter类的WriteLine方法 实现代码: private void button2_Click(object sen ...
    
			
    5. 
						
  5. C++判断两个double类型双精度浮点数是否同号
			
    看到的一种整数的方法 != y < ) 由此, double x,y; == fabs( ) { } 目前想到的比较合适判断方法. 此外这里还有一种强制转换类型求符号位的方法. /** * Ge ...
    
			
    6. 
						
  6. Linux文件的IO操作 一
			
    系统调用 系统调用: 操作系统提供给用户程序调用的一组“特殊”接口,用户程序可以通过这组“特殊”接口来获得操作系统内核提供的服务 为什么用户程序不能直接访问系统内核提供的服务 为了更好地保护内核空间, ...
    
			
    7. 
						
  7. C++ static关键字
			
    一.面向过程中的static 1.修饰全局变量(静态全局变量) (1)静态全局变量在全局数据区分配内存: (2)未经初始化的静态全局变量会被程序自动初始化为0: (3)静态全局变量在申明它的整个文件是 ...
    
			
    8. 
						
  8. js 两个数组对象根据账号比较去重，解决直接splice后数组索引改变
			
    目的获取Arr2中不包含在arr1中的对象 根据Account进行比较,如果相等则删除tempArr数组对象. 结果返回张三 var arr1=[{"account":" ...
    
			
    9. 
						
  9. idea下使用码云插件进行git提交
			
    1)下载插件 file->setting->plugins->右侧搜索gitee->安装->重启ide 2)配置版本控制 file->setting->Ver ...
    
			
    10. 
						
  10. linux or msys2设置网络代理
			
    在文件 .bashrc 中添加 export http_proxy="proxy IP:port" 如 export http_proxy="192.168.0.1:80 ...
    
			
    11.