Skip to content

v0.1.3
Compare
Choose a tag to compare
@github-actions github-actions released this 15 Oct 19:22
· 983 commits to main since this release
v0.1.3
8bdb195

What's new

Added 🎉

  • Added the ability to track a metric other than "loss" for validation in TorchTrainStep ("torch::train").

Fixed ✅

  • Final model returned from TorchTrainStep ("torch::train") will have best weights loaded.
  • Checkpoints are saved from TorchTrainStep ("torch::train") even when there is no validation loop.
  • Fixed TorchTrainStep ("torch::train") when validation_split is None.
  • Fixed distributed training with TorchTrainStep ("torch::train") on GPU devices.

Commits

ba05e79 Torch train updates and distributed fixes (#27)

Assets 4
Loading