we should modify train.py file by this code:

freeze = ['model.%s.' % x for x in range(10)] # parameter names to freeze (full or partial) if any(freeze): for k, v in model.named_parameters(): if any(x in k for x in freeze): print('freezing %s' % k) v.requires_grad = False