fix
This commit is contained in:
parent
74117e2a3f
commit
7faa3c4b1c
@ -229,7 +229,7 @@ class ModelArguments:
|
||||
|
||||
def freeze_model_weights(model: torch.nn.Module) -> None:
|
||||
count = 0
|
||||
print(len(model.parameters())
|
||||
print(len(model.parameters()))
|
||||
for param in model.parameters():
|
||||
count += 1
|
||||
if count <= 10:
|
||||
|
Loading…
Reference in New Issue
Block a user