stared / livelossplot

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

Training pauses on epoch end and won't continue until window is closed #141

Open JeffStodd opened 2 months ago

JeffStodd commented 2 months ago

Same issue as https://github.com/stared/livelossplot/issues/76#issue-539629345

The training stops at the end of the epoch. I have to close the window to start training on the next epoch, and keep repeating this.

    tensorboard_callbacks = []
    if plot:
        tensorboard_callbacks = [PlotLossesKerasTF()]

    model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
    model.fit(normalizedInput.values, expectedOutput.values, validation_data=(normalizedValidationInput.values, expectedValidationOutput.values), epochs=100, batch_size=8, callbacks=tensorboard_callbacks)

Running as script on Windows Python 3.9.13 tensorflow 2.13.1 livelossplot 0.5.5