stared / livelossplot

Live training loss plot in Jupyter Notebook for Keras, PyTorch and others
https://p.migdal.pl/livelossplot
MIT License
1.29k stars 142 forks source link

AttributeError: 'PlotLossesCallback' object has no attribute 'on_train_batch_begin' #67

Closed maxouille closed 4 years ago

maxouille commented 5 years ago

Hi,

I'm using this lib (which seems cool btw) and I got this error message : AttributeError: 'PlotLossesCallback' object has no attribute 'on_train_batch_begin'

I'm using TF 2.0 Beta 1 and Keras in python.

Thx !

ucekmez commented 5 years ago

you have to import the library as follows, if you want to use tf.keras:

from livelossplot.tf_keras import PlotLossesCallback