pytorch 中 切换模型更新 与否

一种方式

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）