def freeze()

in code/train.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def freeze(model, frozen_layers):
    modules = [model.bert.encoder.layer[:frozen_layers]] 
    for module in modules:
        for param in module.parameters():
            param.requires_grad = False