From f695c5c5d0b7f2924e6f70b97f2afd1df26e5ca0 Mon Sep 17 00:00:00 2001
From: Zofia Galla <zosiagalla@gmail.com>
Date: Sun, 16 May 2021 12:10:52 +0200
Subject: [PATCH] Fix sacred capture func

---
 ium_zadanie6_training.py | 1 +
 1 file changed, 1 insertion(+)

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