一种方式

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）

标签：与否,enable,module,pytorch,disable,切换,model,gradients
来源： https://blog.csdn.net/weixin_44003119/article/details/122260925