search

pesser / edflow

Reduce boilerplate code for your ML projects. TensorFlow and PyTorch.
https://edflow.readthedocs.io/
MIT License
24 stars 13 forks source link

restore batch step and epoch step #238

Closed pesser closed 4 years ago

pesser commented 4 years ago

when restoring the global step, edflow now also restores the epoch and batch progress bars. note however that this only makes sense if the batch size wasn't changed.

in addition this PR improves the progress bar display a bit. since some new version of tqdm, progress bars accumulate without setting leave=False. This should also be done in callbacks' progress bars. Also we only keep two lines of progress bars now, the epoch always on top and below it the current iteration loop (training or validation or callbacks' iteration).

project and entity can be specified for wandb to log into team accounts, ckpt_zero can be set to True to save the zero-th checkpoint and debug/{disable_integrations, max_examples} can be used to adjust behavior in debug mode.

theRealSuperMario commented 4 years ago

Have we agreed on merging PRs without getting tests to run on the dev branch?

I think at least formatting should be enforced, because it makes reading PRs and diffing easier.

pesser commented 4 years ago

thanks for the reminder. done.