Skip to content

v0.1.0

Compare
Choose a tag to compare
@beekill95 beekill95 released this 03 Nov 02:42
· 51 commits to main since this release

TrainingLoop instances can be initialized directly.
They receive an instance of TrainingStep specifying the logic of training/validating a model.

Removed

  • Breaking: Remove SimpleTrainingLoop, replace it with SimpleTrainingStep.
    (#11)
  • Remove tensorboard dependency.

Added

  • Add usage section in README.
  • Add an example for training conditional GAN on Fashion MNIST with custom loop.