Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Plotting all history points when using fit_generator #7

Closed
OlivierData opened this issue Mar 27, 2018 · 1 comment
Closed

Plotting all history points when using fit_generator #7

OlivierData opened this issue Mar 27, 2018 · 1 comment

Comments

@OlivierData
Copy link
Contributor

Hello,

This is a great alternative tool to Tensorboard for watching deep learning models.

Having said that, I noticed a bug when using fit_generator(). PlotLossesKeras() only plots the last point for each loss/accuracy and you end up with a straight line which is not very usefull...

capture d ecran 2018-03-27 a 19 49 47

Could you test your tool with fit_generator and provide a solution for this method ?

For information, when I plot directly from model.history.history, I get a good result (same thing when using fit method):
capture d ecran 2018-03-27 a 20 00 44

@OlivierData
Copy link
Contributor Author

Eventually I found the problem and here is a workaround :

In
PlotLossesKeras.on_epoch_end()
replace :
self.logs.append(logs)
with :
self.logs.append(logs.copy())

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant