Source code for super_gradients.training.exceptions.sg_model_exceptions
[docs]class UnsupportedTrainingParameterFormat(Exception):
"""Exception raised illegal training param format.
Attributes:
message -- explanation of the error
"""
def __init__(self, desc):
self.message = "Unsupported training parameter format: " + desc
super().__init__(self.message)
[docs]class UnsupportedOptimizerFormat(UnsupportedTrainingParameterFormat):
"""Exception raised illegal optimizer format.
Attributes:
message -- explanation of the error
"""
def __init__(self):
super().__init__(
"optimizer parameter expected one of ['Adam','SGD','RMSProp'], or torch.optim.Optimizer object")