From e6f8958b6aedb7913b2beb0d97f557b88c8e736a Mon Sep 17 00:00:00 2001
From: s444501 <krywas@st.amu.edu.pl>
Date: Sun, 12 Feb 2023 14:54:19 +0100
Subject: [PATCH] fix

---
 run_translation_freezing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run_translation_freezing.py b/run_translation_freezing.py
index bdeb26d..86f7266 100644
--- 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: