一种方式
pytorch中,两个网络怎么固定一个训练另外一个? - 知乎
一般使用的模块代码
def disable_gradients(module):
for p in module.parameters():
p.requires_grad = False
def enable_gradients(module):
for p in module.parameters():
p.requires_grad = True
例子
model = net()
# 关闭更新
disable_gradients(model)
# 开启更新
enable_gradients(model)