AI入门----PyTorch的结构

电玩女神 2022-12-28 14:03 182阅读 0赞

PyTorch中常用的模块的结构如下：

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhbmdlaWwwMDc_size_16_color_FFFFFF_t_70]  
1、torchvision包：  
① datasets：datasets中包含常用的数据集，例如MNIST，CIFA等；  
② models：models中包含一些经典的已经训练好的模型，例如VGG16，VGG19、ResNet等，这些模型主要是用来进行迁移学习的。关于迁移学习，我们将在后续章节中进行介绍。  
③ transforms：把数据集中的内容转化成tensor。

2、torch包中主要包括四块：  
① autograd包。里面有用来定义PyTorch变量的Variable。  
② nn (neural network)包。nn包中包含的内容最丰富：损失函数 (MSELoss, crossEntropyLoss等)， functional (一般用F表示，包含各种激活函数、池化相关函数、正则化相关函数等)，网络层的函数 (线性层/全连接层Linear, 卷积函数Conv2d等)。其中需要注意的是，卷积和池化是在一起使用的，但是却在不同的包里。原因是：PyTorch把需要训练参数的内容都放在网络层，把不需要训练参数的工具函数都放到了functional中。神经网络中的参数w、b和卷积参数都在网络层中，我们要训练的就是这些参数。而无论是激活函数，还是池化 (池化的卷积核大小、步长等可以计算出来)、dropout等都可以没有参数。nn包还有一个Module模块，这个模块中包含了parameters()、train()和eval()等方法。parameters()方法可以直接获取到要训练的所有参数。这些参数必须是在\_\_init\_\_()中用self.conv1，self.fc1等指定的。如果在神经网络模型中使用了dropout，则在训练的时候会随机丢弃掉部分神经元，这就要在训练时用train()方法指定是训练时的神经网络。对应的，如果是已经训练好的神经网络的预测，则要用eval()方法指定是测试用的神经网络。最神奇的是forward()方法，模型的前向传播必须调用forward()，甚至可以不用forward这个名字，直接可以调用，例如：model(input)，反向传播的过程其实就是对forward()方法中的函数进行反向求导。  
③ optim包。里面有梯度下降相关的函数，例如SGD、Adam等。(Momentum作为一个参数被包含在SGD中了)。还有梯度清零 (zero\_grad)、进行一次优化 (step) 等函数。  
④ torch模块下属有众多处理tensor的方法。  
根据上述内容，我们至少可以实现全连接的神经网络、卷积神经网络以及迁移学习了，是不是很期待实现这些功能？下面我们就开始从PyTorch的基本数据类型、基本操作开始讲解。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhbmdlaWwwMDc_size_16_color_FFFFFF_t_70]: /images/20221120/4442334fbca64b59962ed245d13b1dc9.png