Releases: beekill95/torch-training-loop
Releases · beekill95/torch-training-loop
v0.1.3
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
Changed
- Support dataloaders without
__len__
implementation. (#21)
Added
- Different verbose levels for training loops'
fit()
function. (#20)
v0.1.1
Added
- Implement distributed training loop. (#12)
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 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
Callbacks will now have references to both training loop and model instances.
v0.0.2
- Fix: Cuda training not working;
- Add: Update progress bar after an epoch finishes.