人脸识别模型的最后完善

                              超可❤的编程知识🌠🌙


经历了双十一剁手的我,现在在掉这笔敲代码,你信吗?

       我狮虎建议我保存权重,不要保存模型,因为保存模型的模型不能很方便的加载。而保存权重的话,没有什么弊端。有一种方法既保存模型又保存权重,这个方法很复杂,但是需要先定义模型再加载。

      我决定,先看一下我狮虎写的代码,自己摸索一下。(悄咪咪的补充一句,你要是想在服务器上面的notebook上面打开一个.py文件啊,你就,上传,就行了(/≧▽≦)/)


我以为这就是保存参数,其实这是在定义优化器,这个优化器对不同的层,设置不一样的lr等


果然是,我狮虎的模型定义的就是fc层和resnet层


那我猜测,及时这里是在仅保存参数了。

下面请看与我的保存模型的代码对比:


model1是我修改后的模型~
这一步我将我的模型model1保存为名为“resnet50_face”的.py文件



出现这个问题的原因是因为模型没有用cuda()


重新运行后,并没有什么效果,还是原来的错误


但是还是有问题

我狮虎说:原来是因为数据没有cuda






终于好了。

继续完善vgg16:


这个问题是我多写了一个vgg16.classifier.这个.classifier是多余的


这是原来的classifier

我做完以后却只剩下:


所以决定,重新写一下这个分类层


每天早上叫醒我的不是梦想,而是,我要继续来实验室完善我的代码……

未完,待续……

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容