본문 바로가기

연구/pytorch

[model freeze] layer 의 일부만 freeze 하기

반응형
    if config.model.use_param_regressor:
        print(f"freeze all parameters except ParamRegressor")
        for name, param in model.named_parameters():
            if 'ParamRegressor' in name:
                # print(f"{name} requires_grad = True")
                pass
            else:
                param.requires_grad = False
                # print(f"{name} requires_grad = False")

 

반응형