pytorch
文章平均质量分 91
holzli
这个作者很懒,什么都没留下…
展开
-
5. nn.Module
torch.nnnn.Parameter:张量子类,表示可学习参数,如weight,biasnn.Module:所有网络层基类,管理网络属性nn.functional:函数具体实现,如卷积,池化,激活函数等nn.init:参数初始化方法nn.Module·parameters:存储管理nn.Parameter类(包括权值、偏置等)·modules:存储管理nn.Module类(管理各...原创 2020-04-28 10:41:59 · 768 阅读 · 1 评论 -
4. Transform
transformstorchvision.transforms:常用的图像预处理方法1. transforms.Normalize(mean, std, inplace=False)功能:逐channel的对图像进行标准化output = (input - mean) / stdmean:各通道的均值std:各通道的标准差implace:是否原地操作2. transforms....原创 2020-04-25 14:08:45 · 306 阅读 · 0 评论 -
深度学习pytorch之一步导入自己的训练集
DataLoader构建可迭代的数据装载器DataLoader(dataset, batch_size=1, shuffle=False, num_works=0)dataset:Dataset类,决定数据从哪儿读取及如何读取batch_size:批大小shuffle:每个epoch是否乱序num_works:是否多进程读取数据DatasetDataset抽象类,所有自定...原创 2020-04-23 17:16:08 · 6556 阅读 · 2 评论 -
2. Autograd
对于torch.Tensor,如果设置属性.requires_grad = True。使用.backward()方法进行自动求取梯度,张量的梯度会保存到.grad属性中。使用.backward()计算梯度。如果张量是一个标量(只有一个元素),不需要对.backward()指定参数;如果张量不止一个元素,需要设定参数,匹配张量的维度。例:a = torch.tensor([1.],requi...原创 2020-04-23 16:00:28 · 133 阅读 · 0 评论 -
1. Tensor
张量是什么?张量是一个多维数组,它是标量、向量、矩阵的高维扩展。张量的属性:data:被包装的tensorgrad:data的梯度grad_fn:创建Tensor的Functionrequires_grad:指示是否需要梯度is_leaf:指示是否是叶子节点dtype:张量的数据类型shape:张量的形状device:张量所在设备张量的创建torch.tensor(dat...原创 2020-04-23 15:55:30 · 134 阅读 · 0 评论