随笔分类 - Pytorch
手写DCGAN
摘要://加上了注释,对pytorch又加深了理解import torch as t from torch import nn from torch.autograd import Variable from torch.optim import Adam from torchvision import transforms from torchvision.utils import make_gri...
阅读全文
Pytorch CNN的各种参数
摘要:conv :卷积层(2d就是二维平面的) kernel_size 卷积核大小 stride 每次移动的步长 padding 四周填充的大小,注意是四周所以在算下一层的向量维度时要将padding 乘以 2 maxpool2d 二维平面的池化层 dense:全链接层 Linear 一层神经元 输入维数
阅读全文
Pytorch 一些函数用法
摘要:PyTorch中view的用法:https://blog.csdn.net/york1996/article/details/81949843 max用法 tensor([[1., 3.], [2., 4.]]) (tensor([2., 4.]), tensor([1, 1]))(tensor([
阅读全文
windows下配置cuda9.0和pytorch
摘要:今天看了看pytorch官网竟然支持windows了,赶紧搞一个。 下载cuda 9.0 https://developer.nvidia.com/cuda-downloads 下载anaconda3 https://www.anaconda.com/download/ (记得安装在默认位置,不然的
阅读全文
Pytorch入门笔记
摘要:__ Net( (conv1): Conv2d(1, 6, kernel_size=(5, 5), stride=(1, 1)) (conv2): Conv2d(6, 16, kernel_size=(5, 5), stride=(1, 1)) (fc1): Linear(in_features=4
阅读全文
浙公网安备 33010602011771号