Pytorch
变量命名好难啊
Master Candidate in FDU
展开
-
Pytorch学习笔记(1):pytoch中如何加载训练数据
1.为什么不需要自己写加载方法 pytorch中提供了两个类用于训练数据的加载,分别是 torch.utils.data.Dataset和 torch.utils.data.DataLoader 。不像torchvision中集合了很多常用的计算机视觉的常用数据集,作为在音乐信息检索这方面,数据集要自己设计加载方法。如果每次不同的数据集都要自己写函数加载, 每次读取代码不能够重用,不同的数据读取...原创 2019-03-29 21:48:00 · 949 阅读 · 0 评论 -
Pytorch学习笔记(2): 一维卷积, RNN, LSTM详解
一维卷积 torch.nn.Conv1d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode=‘zeros’) 这个函数用来对输入张量做一维卷积 in_channel和out_channel是卷积核个数 kernel_size是卷...原创 2019-06-19 15:35:05 · 8947 阅读 · 1 评论 -
pytorch学习笔记(3) 动态调整学习率和如何提取CNN中间层的输出
pytorch学习笔记(3) 动态调整学习率和hook函数的使用. 动态调整学习率 def adjust_learning_rate(optimizer, epoch, params): lr = params["LEARNING_RATE"] * (0.1 **(epoch // 20)) for param_group in optimizer.param_groups: ...原创 2019-07-23 00:23:53 · 1444 阅读 · 1 评论