File size: 95 Bytes
7e213c7
 
 
 
 
 
1
2
3
4
5
6
7


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