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

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

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

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




















