adde info log for training end

apax-hub · Apr 4, 2024 · e59b5d4 · e59b5d4
1 parent 3a81b71
commit e59b5d4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/apax/train/run.py b/apax/train/run.py
@@ -147,3 +147,4 @@ def run(user_config, log_level="error"):
         disable_batch_pbar=config.progress_bar.disable_batch_pbar,
         is_ensemble=config.n_models > 1,
     )
+    log.info("Finished training")