赞
踩
pip install scipy -i https://pypi.douban.com/simple/
pretrained为True时需要下载,在imageNet上训练
(0.15之后改成weights了)
利用现有网络,改动结构
很多框架都利用VGG16提取特征,然后再VGG16后面再加一些网络结构
当前目标:将VGG16改成10中,利用CFAR10训练。
可以加一个线性层,让in_feature=1000,out_feature=10
这种做法只适合前后部分的修改,想要跳层或者跨层连接的话,就只能自己写
或者加载classifier里面
或者直接在里面那个Linear里面改
保存了网络模型中的网络结构和参数
方法一·加载模型
方法二(官方推荐)
输出字典格式
恢复成网络模型:新建网络模型结构
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。