神经网络骨架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

保存了网络模型中的网络结构和参数
-
方法一·加载模型

-
方法二(官方推荐)
输出字典格式

恢复成网络模型:新建网络模型结构

博客围绕神经网络展开,介绍了神经网络骨架nn.Module的继承与初始化,给出scipy的下载命令。还讲述现有网络模型VGG16的使用及修改,如利用其提取特征并改动结构,以适配CFAR10训练。最后提及模型保存的两种方法及加载方式。
624

被折叠的 条评论
为什么被折叠?



