new release

stared · Mar 22, 2018 · fb20506 · fb20506
1 parent 8abe5f4
commit fb20506
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/livelossplot/keras_plot.py b/livelossplot/keras_plot.py
@@ -46,9 +46,10 @@ def on_train_begin(self, logs={}):
         else:
             # the most typical scenario
             losses = [self.model.loss]
-                    
+
         for loss in losses:
             loss_name = loss2name(loss)
+            # better support for multiple loss functions...
             self.metric2printable['loss'] = self.metric2printable.get(loss_name, loss_name) + " (cost function)"
 
         self.max_epoch = self.params['epochs'] if not self.dynamic_x_axis else None

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 setup(name='livelossplot',
-      version='0.1.2',
+      version='0.1.3',
       install_requires=['matplotlib', 'notebook'],
       description='Live training loss plot in Jupyter Notebook for Keras, PyTorch and others.',
       url='https://github.com/stared/livelossplot',