stared/livelossplot

No attribute '_implements_train_batch_hooks'

PepijnBoers opened this issue · 2 comments

I came across an attribute error, when running a simple fit function that has always worked with livelossplot.

My code:

from livelossplot import PlotLossesKeras
model.fit(X_images, y_images, validation_data=(X_val_images, y_val_images), epochs = 50, callbacks=[PlotLossesKeras()])

This gave the following error:
AttributeError: 'PlotLossesCallback' object has no attribute '_implements_train_batch_hooks'

Code runs fine without the callback.

Error caused by using Tensorflow version 2. Issue resolved in Google Colab by running following code before importing Tensorflow:

%tensorflow_version 1.x

are there any plans on making this work with version 2 of tensorflow?