Solved bug closes #33

pab1s · Jul 15, 2024 · 7e8346f · 7e8346f
1 parent 4e26b57
commit 7e8346f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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