From 7e8346fdb375d4dba2b8c29f77953e093637bb0d Mon Sep 17 00:00:00 2001
From: Pablo Olivares <65406121+pab1s@users.noreply.github.com>
Date: Mon, 15 Jul 2024 22:26:34 +0200
Subject: [PATCH] Solved bug closes #33

---
 config/config_test.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/config/config_test.yaml b/config/config_test.yaml
index 8d04935..90abc0f 100644
--- a/config/config_test.yaml
+++ b/config/config_test.yaml
@@ -18,4 +18,4 @@ training:
   batch_size: 64
   num_epochs: 1
   learning_rate: 0.0001
-  freeze_until_layer: classifier.1.weight
+  freeze_until_layer: classifier.1.0.weight