Skip to main content
Version: 0.2

ModelCompileSettings​

CLASS - ModelCompileSettings(

)

dataclass to store all the Deep Learning objects related to the Compression Job.

Class Variables​

  • optimizer (Any) - Optimizer object. In PyTorch it's an object from: torch.optim... When Resuming, 'optimizer' can be set to None unless it is a custom Optimizer.
  • training_losses (Dict[str, object]) - Dictionary mapping between a Name to a Loss Object used for Training. Can be a function that accepts (preds, target) or a Metric that implements an interface like torchmetrics.
  • evaluation_losses (Optional[Dict[str, object]]) - Dictionary mapping between a Name to a Loss Object used for Evaluation. Can be a function that accepts (preds, target) or a Metric that implements an interface like torchmetrics.
  • training_metrics (Optional[Dict[str, object]]) - Dictionary mapping between a Name to a Metric Object used for Training. Can be a function that accepts (preds, target) or a Metric that implements an interface like torchmetrics.
  • evaluation_metrics (Optional[Dict[str, object]]) - Dictionary mapping between a Name to a Metric Object used for Evaluation. Can be a function that accepts (preds, target) or a Metric that implements an interface like torchmetrics.