2019年06月_samylee

07月 06月 05月 04月 03月 02月 01月

原创 PyTorch-GPU加速

PyTorch-GPU加速硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识将数据和网络都推到GPU，接上.cuda()二、代码展示import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvis...

2019-06-12 13:35:03 9982 2

原创 PyTorch-初遇卷积神经网络

PyTorch-初遇卷积神经网络硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识MNIST数据测试分类二、代码展示import os# third-party libraryimport torchimport torch.nn as nnimport torch.utils.dat...

2019-06-06 14:45:02 1191

原创 PyTorch-加速网络训练

PyTorch-加速网络训练硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、SGD+Momentum2、RMSprop3、Adam二、代码展示import torchimport torch.utils.data as Dataimport torch.nn.fun...

2019-06-05 16:59:24 1171

原创 PyTorch-数据批处理

PyTorch-数据批处理硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、学习TensorDataset2、学习DataLoader二、代码展示import torchimport torch.utils.data as Datatorch.manual_seed(1) #...

2019-06-04 15:25:37 1203

原创 PyTorch-模型保存与提取

PyTorch-模型保存与提取硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、save有两种形式，一种是全图保存，一种是参数保存2、load也有两种形式，一种是全图加载，无需单独建立模型，一种是参数加载，但是需要单独建立模型二、代码展示import torchimport matpl...

2019-06-03 15:43:35 1358

原创 PyTorch-Sequential网络快速搭建法

PyTorch-Sequential网络快速搭建法硬件：NVIDIA-GTX1080软件：Windows7、python3.6.5、pytorch-gpu-0.4.1一、基础知识1、torch.nn.Sequential()方法与class Net方法有相同功效，且Sequential方法更简单明了！2、以Linear关系分类为例（https://blog.csdn.net/s...

2019-06-01 17:37:52 703