CODE EXAMPLE FOR PYTHON

load model pytorchand freeze

for param in model*.parameters():
    param.requires_grad = False