v0.1.0
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 withSimpleTrainingStep
.
(#11) - Remove
tensorboard
dependency.
Added
- Add usage section in README.
- Add an example for training conditional GAN on Fashion MNIST with custom loop.