神经网络骨架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
保存了网络模型中的网络结构和参数
-
方法一·加载模型
-
方法二(官方推荐)
输出字典格式
恢复成网络模型:新建网络模型结构