transfer for real

francesco-vaselli · Aug 29, 2023 · c515d17 · c515d17
1 parent 5844b2d
commit c515d17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/models/train_model.py b/src/models/train_model.py
@@ -360,8 +360,8 @@ def main():
         learning_rate,
         model_config,
         log_name,
-        save_model=True,
-        transfer_learning=False,
+        save_model,
+        transfer_learning,
     )