Skip to content

Releases: beekill95/torch-training-loop

v0.1.3

Choose a tag to compare

@beekill95 beekill95 released this 22 Dec 08:04

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

Choose a tag to compare

@beekill95 beekill95 released this 27 Nov 07:29

Changed

  • Support dataloaders without __len__ implementation. (#21)

Added

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

v0.1.1

Choose a tag to compare

@beekill95 beekill95 released this 11 Nov 07:37

Added

  • Implement distributed training loop. (#12)

v0.1.0

Choose a tag to compare

@beekill95 beekill95 released this 03 Nov 02:42

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

Choose a tag to compare

@beekill95 beekill95 released this 25 Oct 07:51
4dc2c9c

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

v0.0.2

Choose a tag to compare

@beekill95 beekill95 released this 25 Oct 03:58
b2e7e51
  • Fix: Cuda training not working;
  • Add: Update progress bar after an epoch finishes.

v0.0.1

Choose a tag to compare

@beekill95 beekill95 released this 24 Oct 07:18
e68f506

Initial release.