Adding decorators for training progress bars, etc. (#79)
* Adding decorators
* Rewrote decorators.
* Adding demo
* Added test.
* Added the decorators.
* Added docstring
* Bumping version.
* Default step_fn
* Explicitly set tracker step.
* Moving step
* Adding step to init
* dict, not defaultdict when saving history
* Moving lambda outside.
* Changing torchmetrics.
* making dummy
* Device
* return self
* missing item
* No more torchmetrics
* Load step
* Allow writer to be None
* Moving write call
* Moving ddp active to arguments of track.
* Longer width
* skip nans in taking mean.
* Fixing mean.
* Cleaning up decorators a bit.
* Adding docstrings
* Should update even if not dict, for pbars.
* Fixing issue with dropping float/int.
---------
Co-authored-by: pseeth <[email protected]>
