CODE EXAMPLE FOR PYTHON

pytorch freeze layers

# freeze all layers... but you get the point
for param in model.parameters():
    param.requires_grad = False