feat(Config): added flag to save models

2026-08-22 07:18:08 +00:00 · 2022-02-19 15:06:05 +01:00
parent 77206a5d0a
commit 7db1c4dc00
3 changed files with 5 additions and 1 deletions
@@ -35,7 +35,8 @@ def run_pipeline(
        wandb,
        sweep,
    )
-    save_models(pipeline_outcome, config)
+    if config.save_models:
+        save_models(pipeline_outcome, config)
    return pipeline_outcome, config