Pytorch
Vivianyzw
https://github.com/Vivianyzw
展开
-
Pytorch学习第一讲:数据加载
Pytorch官网上给出了一个关于加载CIFAR10数据集的例子:主要使用了torchvision数据包,里面有一些ImageNet,CIFAR-10和MNIST等常见数据集。加载数据分成三个部分:torchvision.transforms(数据预处理)torchvision.datasets(数据集读取)torchvision.DataLoader(数据集加载)代码实现如下:import to...原创 2018-07-09 14:23:08 · 976 阅读 · 0 评论 -
Pytorch学习第二讲:网络创建
Pytorch官网的例子是一个LeNet网络:import torch.nn as nnimport torch.nn.functional as Fclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5)...原创 2018-07-09 14:48:04 · 1788 阅读 · 0 评论 -
Pytorch学习第三讲:训练网络
Pytorch官网也给了训练网络实例:for epoch in range(2): # loop over the dataset multiple times running_loss = 0.0 for i, data in enumerate(trainloader, 0): # get the inputs inputs, labels =...原创 2018-07-09 15:35:01 · 1351 阅读 · 1 评论 -
Pytorch学习第四讲:加载预训练模型
东风的地方1. 直接加载预训练模型在训练的时候可能需要中断一下,然后继续训练,也就是简单的从保存的模型中加载参数权重:net = SNet()net.load_state_dict(torch.load("model_1599.pkl"))这种方式是针对于之前保存模型时以保存参数的格式使用的:torch.save(net.state_dict(), "model/model_1599.pkl")...原创 2018-07-16 11:04:58 · 16791 阅读 · 6 评论 -
Pytorch学习第五讲:LSTM网络实现
这里主要记录一下lstm网络的pytorch实现:import torchimport torch.nn as nnfrom torch.autograd import Variableimport torch.nn.functional as Fclass my_lstm(nn.Module): def __init__(self): super(my_lst...原创 2018-07-16 14:48:03 · 3671 阅读 · 0 评论 -
Pytorch学习第六讲:实现裁剪后的resnet50
先上一个pytorch里自带的一个版本这个版本实现的比较简洁优雅,但如果你想自己改变一些参数,就比较难调,所以这篇文章,会基于该版本中最基本的一个bottleblock来做一个自己实现的版本,这个版本更方便修改参数,调试等。import torch.nn as nnimport torch.nn.functional as Fimport mathimport torchdef c...原创 2019-07-22 17:43:43 · 2511 阅读 · 5 评论