[Freeze Until] #python #pytorch

freeze_until.py

# freezing layers !!!
#-------------------------------------------------------------------------------
def freeze_until(net, param_name):
    found_name = False
    for name, params in net.named_parameters():
        if name == param_name:
            found_name = True
        params.requires_grad = found_name
        
# freeze_until(net, "layer3.0.conv1.weight")