人工智能学习06--pytorch06--神经网络骨架nn.Module & scipy下载 & 现有网络模型的使用及修改(VGG16)

神经网络骨架nn.Module

在这里插入图片描述

  • 括号里nn.Module表示继承Module类
  • init 初始化
    调用父类初始化函数
  • forward
    在这里插入图片描述

scipy下载

pip install scipy -i https://pypi.douban.com/simple/
在这里插入图片描述

现有网络模型的使用及修改(VGG16)

在这里插入图片描述
pretrained为True时需要下载,在imageNet上训练
(0.15之后改成weights了)
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
利用现有网络,改动结构
很多框架都利用VGG16提取特征,然后再VGG16后面再加一些网络结构

当前目标:将VGG16改成10中,利用CFAR10训练。

可以加一个线性层,让in_feature=1000,out_feature=10
这种做法只适合前后部分的修改,想要跳层或者跨层连接的话,就只能自己写

在这里插入图片描述
或者加载classifier里面
在这里插入图片描述
或者直接在里面那个Linear里面改
在这里插入图片描述
在这里插入图片描述

模型的保存

  • 方法1

在这里插入图片描述
保存了网络模型中的网络结构和参数

  • 方法一·加载模型
    在这里插入图片描述

  • 方法二(官方推荐)
    输出字典格式
    在这里插入图片描述
    恢复成网络模型:新建网络模型结构
    在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/AMWICD/article/details/128756713