Skip to content

v1.0.6

Choose a tag to compare

@kisonho kisonho released this 24 Oct 15:43

Updates:

  • Add results property for the metrics to return all current results as a torch.Tensor
  • Add threshold settings to MeanIoU metric
  • Introducing FrequencyCallback with frequency support
  • Introducing DynamicWeight callback that dynamically change the weight property in an object
  • Introducing predict method in TestingManager to predict a dataset and returns a list of predictions
  • Parsing device in fit and test as a list of specified devices