diff --git a/apax/train/run.py b/apax/train/run.py
index bc1bdf68..4b0fd77f 100644
--- 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")