From e59b5d4f415a65d3b78ec41960b7dfa96a3fbd60 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Moritz=20R=2E=20Sch=C3=A4fer?=
 <schaefer@theochem.uni-stuttgart.de>
Date: Thu, 4 Apr 2024 09:59:11 +0200
Subject: [PATCH] adde info log for training end

---
 apax/train/run.py | 1 +
 1 file changed, 1 insertion(+)

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")