Skip to content

v1.1

Choose a tag to compare

@kisonho kisonho released this 10 Mar 15:34
· 611 commits to main since this release

API updates:

  • Add softmax input and target control for torchmanager.losses.KLDiv loss
  • Implement MSE loss with NAN replaced to zero
  • Introducing errors package in torchmanager_core
  • Introducing Resulting protocol
  • Introducing torchmanager.callbacks.Experiment callback that combines torchmanager.callbacks.LastCheckpoint callback, multiple torchmanager.callbacks.BestCheckpoint callbacks, and torchmanager.callbacks.TensorBoard callback with formatted logs, which all saved in a wrapped .exp formatted folder, tensorboard package is required
  • Introducing torchmanager.data package
  • Introducing torchmanager.losses.ParallelLoss for loss functions multi-GPUs support
  • Option to replace NAN to zero after log calculation in KLDiv loss
  • Support of listing checkpoints inside an experiment .exp folder using torchmanager.train.list_checkpoints method
  • Support of loading a torchmanager.train.Checkpoint by .exp folder directory and checkpoint name using torchmanager.train.load method

Other updates:

  • Performance improvement
  • Minor bugs fixed