![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Lzj000lzj
这个作者很懒,什么都没留下…
展开
-
pytorch Sequential卷积神经网络(padding)
数据准备import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# Device configurationdevice = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')...转载 2019-07-07 15:52:16 · 1407 阅读 · 0 评论 -
pytorch简单的全连接网络
定义device,方便在所有机器上运行。将模型变量和模型的输入变量转为to devicewith torch.no_grad()开始test模式import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# Device configurationd...转载 2019-07-07 10:36:40 · 3536 阅读 · 0 评论 -
pytorch简单的逻辑回归
import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transforms# Hyper-parameters input_size = 784num_classes = 10num_epochs = 5batch_size = 100learning_rate ...转载 2019-07-07 10:33:13 · 178 阅读 · 0 评论 -
pytorch简单的线性回归
import torchimport torch.nn as nnimport numpy as npimport matplotlib.pyplot as plt# Hyper-parametersinput_size = 1output_size = 1num_epochs = 60learning_rate = 0.001# 拟合一个线性回归,y=w*x+bx_tr...转载 2019-07-06 18:21:38 · 164 阅读 · 0 评论 -
pytorch迁移学习,使用预训练模型
# Download and load the pretrained ResNet-18.resnet = torchvision.models.resnet18(pretrained=True)# 将参数设置为不可修改for param in resnet.parameters(): param.requires_grad = False# 替换网络的顶层resnet.fc...转载 2019-07-06 18:14:05 · 2364 阅读 · 0 评论 -
pytorch常用的数据预处理
DataLoader使用dataloader方便数据取出。定义CustomDataset类方便对接dataloader类型class CustomDataset(torch.utils.data.Dataset): def __init__(self): # TODO # 1. Initialize file paths or a list of f...原创 2019-07-06 17:42:31 · 837 阅读 · 0 评论 -
pytorch梯度
tensor梯度的相关性若一个节点requires_grad被设置为True,那么所有依赖它的节点的requires_grad都为True。0 import torch1 x=torch.ones(1)2 w=torch.ones(1,requires_grad=True)3 y=x*w4 x.requires_grad,w.requires_grad,y.requ...原创 2019-07-06 17:02:12 · 2357 阅读 · 0 评论 -
pytorch RNN实现分类
数据加载(简单看)from __future__ import unicode_literals, print_function, divisionfrom io import openimport globimport osimport torchdef findFiles(path): return glob.glob(path)#print(findFiles('data/n...原创 2019-07-06 15:56:34 · 3937 阅读 · 1 评论 -
pytorch RNN(构建多个相似结构的模型)
加载数据(简单看)from __future__ import unicode_literals, print_function, divisionfrom io import openimport globimport osimport unicodedataimport stringall_letters = string.ascii_letters + " .,;'-"n_...转载 2019-07-05 17:16:50 · 848 阅读 · 0 评论 -
pytorch保存 加载模型
state_dictstate_dict结构通过层去匹配参数张量Define model class TheModelClass(nn.Module): def __init__(self): super(TheModelClass, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5)...转载 2019-07-04 20:29:34 · 224 阅读 · 0 评论 -
pytorch之GPU数据并行
使模型在gpu上运行在原来的代码上修改了两处,如代码标注所示device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")net.to(device)#1.网络参数数据要是GPU格式for epoch in range(2): # loop over the dataset multiple times ...转载 2019-07-04 18:57:01 · 605 阅读 · 0 评论 -
pytorch之简单的卷积神经网络
网络样例import torchimport torch.nn as nnimport torch.nn.functional as F#输入图片的格式是32*32class Net(nn.Module): def __init__(self):#只是定义网络中需要用到的方法 super(Net, self).__init__() # 1 in...转载 2019-07-04 17:14:00 · 1666 阅读 · 0 评论 -
pytorch 张量基本操作
张量初始化构造一个未初始化的张量torch.empty(5,3)构造一个随机初始化的张量 torch.rand(5,3)构造一个指定数据类型的全0张量torch.zeros(5,2,dtype=torch.long)通过以读取的数据构建张量 torch.tensor([5,5,5,5,5])借助已有张量的size构造新的张量 y=torch.randn_like(x,dtype=tor...转载 2019-07-04 14:12:02 · 2876 阅读 · 0 评论 -
pytorch RNN处理文本
数据预处理见 NLP数据预处理 词的编码以及取batchimport torchimport torch.nn as nnimport numpy as npfrom torch.nn.utils import clip_grad_norm_from data_utils import Dictionary, Corpus# Device configurationdevice...转载 2019-07-07 19:29:57 · 886 阅读 · 0 评论 -
pytorch LSTM图像分类
一个问题:pytorch官方文档对LSTM的输入参数的格式是input of shape (seq_len, batch, input_size),但是本例中images.reshape(-1, sequence_length, input_size)的输入格式为batch,seq_len, input_size,是不是官文写错了?import torch import torch.nn...转载 2019-07-07 14:05:52 · 7753 阅读 · 13 评论