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)
上一篇:phpstorm输入中文显示错误


下一篇:使用 composer + phpstorm 为 PHP 单元测试助力