diff --git a/ium_zadanie6_training.py b/ium_zadanie6_training.py
index 5a0e995..d80cd15 100644
--- a/ium_zadanie6_training.py
+++ b/ium_zadanie6_training.py
@@ -21,6 +21,7 @@ def my_config():
     learning_rate = float(sys.argv[1])
 
 
+@ex.capture
 def prepare_train_model(learning_rate, _run):
 	_run.info["prepare_model"] = str(datetime.now())