Skip to content

v1.2

Choose a tag to compare

@kisonho kisonho released this 10 Nov 16:22
· 346 commits to main since this release

API Updates:

  • Add class index control for metrics.BinaryConfusionMetric
  • Add EPOCH_START and BATCH_START to torchmanager_core.protocols.Frequency
  • Add number of workers support for data.Dataset
  • Add return_summary parameter for fit method to return training summary after training finished
  • Basic configurations implementation with configs.Configs class
  • Change metrics.ConfusionMetrics to abstract class
  • Deprecated train package
  • Implements losses.MAE, metrics.FeatureMetric, metrics.ExtractorScore, metrics.FID, metrics.F1, metrics.Precision, metrics.Recall, metrics.SSIM, metrics.Dice and metrics.PartialDice, scipy package is required for FID
  • Introducing backward, forward function to do backward propagation and forward pass in Manager
  • Introducing configs package
  • Introducing random package with freezing/unfreezing random seeds
  • Introducing a new version convert mechanism using torchmanager_core.protocols.VersionConvertible protocol
  • Introducing data_parallel and reset method in Manager for multi-GPUs control
  • Use forward method to calculate TP, FP, and FN and further calculate metric in forward_metric method in metrics.BinaryConfusionMetric

Other updates:

  • Enables update on batch or epoch start for callbacks.FrequencyCallback
  • Improved Exception handling during prediction
  • Minor bugs fixed
  • Removed deprecated methods