超可❤的编程知识🌠🌙
经历了双十一剁手的我，现在在掉这笔敲代码，你信吗？
我狮虎建议我保存权重，不要保存模型，因为保存模型的模型不能很方便的加载。而保存权重的话，没有什么弊端。有一种方法既保存模型又保存权重，这个方法很复杂，但是需要先定义模型再加载。
我决定，先看一下我狮虎写的代码，自己摸索一下。（悄咪咪的补充一句，你要是想在服务器上面的notebook上面打开一个.py文件啊，你就，上传，就行了(/≧▽≦)/）
我以为这就是保存参数，其实这是在定义优化器，这个优化器对不同的层，设置不一样的lr等
果然是，我狮虎的模型定义的就是fc层和resnet层
那我猜测，及时这里是在仅保存参数了。
下面请看与我的保存模型的代码对比：
model1是我修改后的模型~
这一步我将我的模型model1保存为名为“resnet50_face”的.py文件
出现这个问题的原因是因为模型没有用cuda()
重新运行后，并没有什么效果，还是原来的错误
但是还是有问题我狮虎说：原来是因为数据没有cuda
终于好了。
继续完善vgg16:
这个问题是我多写了一个vgg16.classifier.这个.classifier是多余的
这是原来的classifier
我做完以后却只剩下：
所以决定，重新写一下这个分类层
每天早上叫醒我的不是梦想，而是，我要继续来实验室完善我的代码……未完，待续……
网友评论