Skip to content

Releases: beekill95/torch-training-loop

v0.1.3

22 Dec 08:04
Compare
Choose a tag to compare

Fixed

  • Not passing metrics to SimpleTrainingStep won't cause initialization error. (#26)
  • Displaying scientific notation when metrics values becoming too large or too small. (#29)

v0.1.2

27 Nov 07:29
Compare
Choose a tag to compare

Changed

  • Support dataloaders without __len__ implementation. (#21)

Added

  • Different verbose levels for training loops' fit() function. (#20)

v0.1.1

11 Nov 07:37
Compare
Choose a tag to compare

Added

  • Implement distributed training loop. (#12)

v0.1.0

03 Nov 02:42
Compare
Choose a tag to compare

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.

v0.0.3

25 Oct 07:51
4dc2c9c
Compare
Choose a tag to compare

Callbacks will now have references to both training loop and model instances.

v0.0.2

25 Oct 03:58
b2e7e51
Compare
Choose a tag to compare
  • Fix: Cuda training not working;
  • Add: Update progress bar after an epoch finishes.

v0.0.1

24 Oct 07:18
e68f506
Compare
Choose a tag to compare

Initial release.