From cfa88b2b0e98c469dcd2b91a5413a78e4da05128 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Fri, 13 Sep 2019 17:10:50 +0300
Subject: [PATCH] Add example of model configuration dictionary logging

---
 examples/keras_tensorboard.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/keras_tensorboard.py b/examples/keras_tensorboard.py
index ecf137f8..c581a973 100644
--- a/examples/keras_tensorboard.py
+++ b/examples/keras_tensorboard.py
@@ -90,6 +90,8 @@ model.compile(loss='categorical_crossentropy',
 
 # Connecting TRAINS
 task = Task.init(project_name='examples', task_name='Keras with TensorBoard example')
+task.set_model_config(config_dict={'test': 1337, 'nested': {'key': 'value', 'number': 1}})
+
 # Advanced: setting model class enumeration
 labels = dict(('digit_%d' % i, i) for i in range(10))
 task.set_model_label_enumeration(labels)