如何只训练网络中的全连接层

https://blog.csdn.net/weixin_32759777/article/details/105661316
https://dongfangyou.blog.csdn.net/article/details/105661665
参照上面的代码可以将网络的卷积部分使用其他模型参数
全连接层使用自定义的结构
训练的时候让优化器只优化全连接层如下所示

  if optimizer is None:
        optimizer = optim.SGD(model.classifier.parameters(), lr=0.0001, momentum=0

               

上一篇:深度学习编译与优化Deep Learning Compiler and Optimizer


下一篇:pytorch打印当前学习率