Fix sacred capture func

This commit is contained in:
Zofia Galla 2021-05-16 12:10:52 +02:00
parent 02fae476bf
commit f695c5c5d0

View File

@ -21,6 +21,7 @@ def my_config():
learning_rate = float(sys.argv[1]) learning_rate = float(sys.argv[1])
@ex.capture
def prepare_train_model(learning_rate, _run): def prepare_train_model(learning_rate, _run):
_run.info["prepare_model"] = str(datetime.now()) _run.info["prepare_model"] = str(datetime.now())