This commit is contained in:
Andrzej Preibisz 2023-02-12 17:22:29 +01:00
parent 74117e2a3f
commit 7faa3c4b1c

View File

@ -229,7 +229,7 @@ class ModelArguments:
def freeze_model_weights(model: torch.nn.Module) -> None: def freeze_model_weights(model: torch.nn.Module) -> None:
count = 0 count = 0
print(len(model.parameters()) print(len(model.parameters()))
for param in model.parameters(): for param in model.parameters():
count += 1 count += 1
if count <= 10: if count <= 10: