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")