def freeze_network(model): for name, p in model.named_parameters(): if "l0" not in name: p.requires_grad = False return model --- model = freeze_network(model)