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 <prem@descript.com>

Author: pseeth

audiotools/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.6.3"
+__version__ = "0.6.4"
 from .core import AudioSignal
 from .core import STFTParams
 from .core import Meter

audiotools/core/ffmpeg.py

@@ -178,7 +178,7 @@ def load_from_file_with_ffmpeg(cls, audio_path: str, quiet: bool = True, **kwarg
             # offset in a video container.
             pad = ffprobe_offset(audio_path)
             # Don't pad files with discrepancies less than
-            # 0.0.27s - it's likely due to codec latency.
+            # 0.027s - it's likely due to codec latency.
             # The amount of latency introduced by mp3 is
             # 1152, which is 0.0261 44khz. So we
             # set the threshold here slightly above that.

audiotools/ml/__init__.py

@@ -1,3 +1,4 @@
+from . import decorators
 from . import layers
 from . import tricks
 from .accelerator import Accelerator