Version 0.5.4

[0.5.4] - 2023-11-13

Added

• Allow imaging callback's to_file to use state information in the file name

Changed

Deprecated

Removed

Fixed

• Fixed warnings about the epoch argument in schedulers in newer versions of pytorch
• Fixed a bug in access metrics function and callbacks that use it
• Fixed bug where schedulers were called before optimisers with newer versions of pytorch
• Fixed a bug where the csv logger closed the file too early
• Fixed compat with pytorch > 1.1.0 versioning
• Fixed typos in doc strings
• Fixes for tests where pytorch >2 Tensors were causing issues with mocks
• Fix bug in gradient clipping where the parameter generator was consumed on the first pass