0.3.0

Updates

• Adds a new set of callbacks to the Model class that are called at the start/end of each epoch.
• Adds a way to set a custom collate_fn for the dataloader.
• The Model no longer contains abstract methods.
• Changes the call site of model.on_training_start so print statements don't interfere with the progress bar.
• Extend the list of optimizers and schedulers so all the ones provided by PyTorch are registered by default.
• Extend the should_training_stop functionality to allow breaking out of the loop after a training step.
• Adds documentation with Sphinx.
• Allow __version__ to be directly imported from the helios package.

Breaking Changes

• ToTensor has been renamed as ToImageTensor in order to be more explicit about what the class does.

Full Changelog

0.2.0...0.3.0