- 博客(3)
- 收藏
- 关注
原创 pytorch构建模型三种方法
1、首先导入共用的包import torch.nn as nn2、构建模型的四种方法2.1 第一种方法import torch.nn as nn#Method 1 ------------------------------------------------------------------------------class LinearRegression(nn.Mo...
2019-05-24 17:11:20 4212
原创 pytorch模型保存格式
pytorch保存数据保存用到torch.save函数,注意该函数第一个参数可以是单个值也可以是字典,字典可以存更多你要保存的参数(不仅仅是权重数据)pytorch读取数据pytorch读取数据使用的方法和我们平时使用预训练参数所用的方法是一样的,都是使用load_state_dict这个函数。三个核心功能(保存和加载模型)1.torch.save:将序列化的对象保存到disk。这个函数...
2019-05-04 09:14:53 9679 2
原创 pytorch深度学习框架--gpu和cpu的选择
基于pytorch框架,最近实现了一个简单的手写数字识别的程序,在这个过程中,我发现了好多问题。比如,在使用cpu训练的时候就很慢,而且还占内存,要是数据集很大的话,内存不够怎么办呢于是,突发奇想,使用gpu啊,试了一下,果真速度快了不是一丁半点。当然在gpu选择的时候也是可以控制的,你可以使用一个gpu,你也可以使用两个gpu,这个的话完全看你电脑的配置了。对于我的话我选择了两个gpu并行操作,...
2019-04-30 21:23:47 22346 7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人