From 8a4f74ad5056701556f87a08a2f6f285cb54d718 Mon Sep 17 00:00:00 2001 From: lukas leufen <l.leufen@fz-juelich.de> Date: Fri, 19 Jun 2020 16:50:00 +0200 Subject: [PATCH] test update --- test/test_modules/test_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_modules/test_training.py b/test/test_modules/test_training.py index 66ba0709..eb5dfe5a 100644 --- a/test/test_modules/test_training.py +++ b/test/test_modules/test_training.py @@ -156,7 +156,7 @@ class TestTraining: obj.data_store.set("model_path", model_path, "general") obj.data_store.set("model_name", os.path.join(model_path, "test_model.h5"), "general.model") obj.data_store.set("batch_size", 256, "general") - obj.data_store.set("epochs", 2, "general.model") + obj.data_store.set("epochs", 2, "general") clbk, hist, lr = callbacks obj.data_store.set("callbacks", clbk, "general.model") obj.data_store.set("lr_decay", lr, "general.model") -- GitLab