网络冻结

William.csj8 个月前
人工智能·pytorch·神经网络·网络冻结·冻结
Pytorch——训练时，冻结网络部分参数的方法要固定训练网络的哪几层，只需要找到这几层参数(parameter)，然后将其 .requires_grad 属性设置为 False 。然后修改优化器，只将不被冻结的层传入。