Nathan

[FEATURE]: Model Saving and Checkpointing

1

**Is your feature request related to a problem? Please describe.** Model saving and checkpointing is currently disabled for `experiment/run.py` This was due to old pickling errors and the extensive use...

[DOCS]: Add Experiment Examples

1

**Is your feature request related to a problem? Please describe.** The current examples are very limited and only show how to use `disent`. **Describe the solution you'd like** Add examples...

[FEATURE]: Allow Metrics To Directly Accept Frameworks & Datasets

1

**Is your feature request related to a problem? Please describe.** Current metrics require that you provide a representation function. This is inconvenient and always repeated. Metrics also always require that...

Documentation is missing key framework features - augmentations - schedules - creating your own framework - creating your own models - creating your own datasets - visualisations

Improve Tests

2

Tests are currently lacking across disent - data - datasets - schedule - metrics - transformations

The betatcvae implementation is definitely not correct. - loss scaling is not implemented - sane defaults for config

The InfoVAE implementation is probably not correct. - loss scaling might not be correctly implemented - sane defaults for config - irq kernel was removed

The DipVAE implementation is probably not correct. - loss scaling is not be correctly implemented - sane defaults for config

Standardise Loss Reduction Modes

1

Removed `reduction="sum"` in #12, but considerable effort is still required to maintain and verify models (#3, #4, #5) Two options exist: 1. Allow models to specify their allowed modes: `"mean"`...

**Describe the bug** Schedules may unintentionally be affected by the use of validation and test data enabled in release v0.3.4 5695747c1e94420c024f1505d9b8a4b3c81ad610 - These steps might increment trainer.global_step Conversely validation and...