基于pytorch的lenet冻结

基于pytorch的lenet冻结
lenet网络结构
在这里插入图片描述
直接在训练前加入以下步骤:
1、优化器传入所有的参数
在这里插入图片描述
2、冻结所以的卷积层和池化层
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/njj15760675213/article/details/128174439
今日推荐