Source code for super_gradients.training.exceptions.dataset_exceptions


[docs]class IllegalDatasetParameterException(Exception): """ Exception raised illegal dataset param. Attributes: message -- explanation of the error """ def __init__(self, desc): self.message = "Unsupported dataset parameter format: " + desc super().__init__(self.message)
[docs]class EmptyDatasetException(Exception): """ Exception raised when a dataset does not have any image for a specific config Attributes: message -- explanation of the error """ def __init__(self, desc): self.message = "Empty Dataset: " + desc super().__init__(self.message)
[docs]class UnsupportedBatchItemsFormat(ValueError): """Exception raised illegal batch items returned from data loader. Attributes: message -- explanation of the error """ def __init__(self): self.message = "Batch items returned by the data loader expected format: \n" \ "1. torch.Tensor or tuple, s.t inputs = batch_items[0], targets = batch_items[1] and len(" \ "batch_items) = 2 \n" \ "2. tuple: (inputs, targets, additional_batch_items)" super().__init__(self.message)