ModelCompileSettings
CLASS - ModelCompileSettings(
- optimizer:
Any
- training_losses:
Dict[str, object]
- evaluation_losses:
Optional[Dict[str, object]]
= None- training_metrics:
Optional[Dict[str, object]]
= None- evaluation_metrics:
Optional[Dict[str, object]]
= None)
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.