![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 81
geiyes
这个作者很懒,什么都没留下…
展开
-
2021-6-19#MNIST Handwritten Digit Recognition in PyTorch
对照笔记传送门这里写目录标题设置环境准备数据集搭建网络模型训练评估模型性能继续训练设置环境使用Pytorch训练卷积神经网络来识别 MNIST的手写数字首先import 是导入模块的语法,导入torch,torchvision模块:import torchimport torchvision准备数据集导入模块之后,我们就可以接着准备需要使用的数据了。再次之前先定义超参数(Hyperparameter)n_epochs = 3原创 2021-06-19 21:05:45 · 316 阅读 · 0 评论 -
笔记#2021-6-18#pytorch
torch.backends.cudnn.enabled = FalsecuDNN非确定性算法torch.backends.cudnn.enabled = False #禁用torch.backends.cudnn.enabled = True #打开若将其打开,那么cuDNN使用的非确定性算法就会自动寻找最适合当前配置的高效算法,来达到优化运行效率的问题一般来讲,应该遵循以下准则:如果网络的输入数据维度或类型上变化不大,设置 torch.backends.cud原创 2021-06-18 16:12:42 · 369 阅读 · 0 评论 -
pytorch中index_select(),masked_select(),gather()使用
index_select()index_select(input, dim, index) 在指定维度dim上选取,比如选取某些行、某些列input中输入的一个该是一个张量(tensor)dim代表选取的维度:0代表行,1代表列后面的张量代表的是指定的行或列input.index_select(dim, index)此写法与index_select(input, dim, index)效果相同x = torch.tensor([[2,1,3,0],[1,2,0,4],[4,3,2,1]])p原创 2021-05-18 14:35:32 · 2989 阅读 · 0 评论