Changes inner progressbar from counting batches to counting samples.

bstriner / keras-tqdm

Keras integration with TQDM progress bars

MIT License

348 stars 41 forks source link

Changes inner progressbar from counting batches to counting samples. #6

Closed rohitrawat closed 7 years ago

rohitrawat commented 7 years ago

Fixes #5 for cases with variable or unknown batch size (missing params['batch_size']) when using fit_generator().

rotu commented 7 years ago

Works for me! 👍

JingyuQian commented 7 years ago

Works for me too. I previously suffered from disconnected socket error when using Jupyter Notebook. After using keras_tqdm the problem disappears. Additionally what @rohitrawat suggested solves fit_generator problem perfectly. Thank you.

bstriner commented 7 years ago

@rohitrawat Thanks a lot! Sorry it's been a busy month. Will try to release an updated version soon.

Cheers