Helios-ml

Update
* Adds a flag to disable the printing of the banner.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.8...0.1.9

Updates
* Allow easy access to the datasets held by the `DataModule`. Previously there was no direct way of accessing them without having to go through the private members of the `DataModule`. This complicated certain cases where the length of the dataset was required.
* Added a way to halt training based on arbitrary conditions. The main use-case for this is to allow the `Model` sub-classes to halt training when the trained network has converged to a value or if the network is diverging and there's no reason to continue.
* Addresses a potential crash that occurs whenever training occurs on a `None` checkpoint path.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.7...0.1.8

Updates
* For iteration training, the global iteration is now updated correctly. Previously it was updated in the middle of the training loop, which caused the progress bar and the log flag passed in to the model after the batch was over to be out of sync with the global iteration count. This has now been addressed by updating the global iteration count at the top of the iteration loop.
* Removes the callback system from the trainer. Given the current implementation, there's nothing that the callbacks could do that couldn't be performed by overriding the corresponding function in the model or the datamodule.
* Adds wrappers for printing which allow the user to choose which rank (global or local) the print should happen on.
* Adds a wrapper for `torch.distributed.barrier` which works in both distributed and regular contexts.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.6...0.1.7

Updates
* Adds a getter for `swa_utils.EMA` so the underlying network can be easily retrieved.
* Better import support for the core package. Aliasing the core package and importing sub-modules from it is now supported.
* The trainer no longer prints duplicate messages when using distributed training.
* Allows the trainer to populate the registries itself. This provides better support for distributed training that uses spawn.
* The internal distributed flag for the trainer is now correctly set when invoked through `torchrun`.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.5...0.1.6

Updates

* Fixes an issue where printing/saving were incorrectly called whenever training by iteration used accumulation steps. This was caused by an incorrect guarding of the printing, validation, and saving operations.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.4...0.1.5

Updates

* Added a context manager to disable cuDNN benchmark on scope.
* Fixes an issue where cuDNN is disabled upon entering the validation code but is never re-enabled. This could lead to poor performance after the first validation cycle.

Full Changelog
https://github.com/marovira/helios-ml/compare/0.1.3...0.1.4