基于pytorch的lenet冻结

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

你可能感兴趣的:(pytorch,深度学习,python)