![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch深度学习
糖果人碎了
这个作者很懒,什么都没留下…
展开
-
PyTorch学习笔记(二)卷积神经网络流程
深度学习框架的一般流程:1.数据加载和处理2.定义一个模型(包括网络、损失函数和优化器)3.在训练集上进行训练4.在测试集上测试模型的性能(步骤1、2可以调换)步骤一:使用Torchvision加载和规范化CIFAR10训练集和测试集torchvision模块可以加载数据,也可以处理数据等等import torchimport torchvisionimport torchvision.transforms as tranforms#步骤一:使用Torchv原创 2021-11-25 14:29:34 · 774 阅读 · 0 评论 -
PyTorch学习笔记(一)简化的神经网络流程
如何使用PyTorch定义神经网络首先是模型层要用到的一些基本函数#模型层input_image = torch.rand(3,28,28) #随机生成三个28*28的为输入数据print(input_image.size())#nn.Flatten层#将28*28的数据转化为784的一维数据flatten = nn.Flatten()#要先初始化flatten层 flat_image = flatten(input_image)#调用nn.Flatten层print(fla.原创 2021-11-24 19:45:50 · 1232 阅读 · 1 评论