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:
|
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:
|
||||||
|
Loading…
Reference in New Issue
Block a user