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)