fix

2023-02-12 14:54:19 +01:00 · 2023-02-12 14:54:19 +01:00 · e6f8958b6a
commit e6f8958b6a
parent 99b93a2579
1 changed files with 1 additions and 1 deletions
--- a/run_translation_freezing.py
+++ b/run_translation_freezing.py
@ -261,7 +261,7 @@ def freeze_model_weights(model: torch.nn.Module) -> None:
    count = 0
    for param in model.parameters():
        count += 1
-        if count < 20:
+        if count <= 20:
            logger.info(f'Freezing layer {count}')
            param.requires_grad = False
        else: