check_dataset#

deepinv.datasets.check_dataset(dataset, allow_non_tensor=True)[source]#

Check that a torch dataset is compatible with DeepInverse.

For details of what is compatible, see ImageDataset.

Parameters:
  • dataset (torch.utils.data.Dataset) – torch dataset.

  • allow_non_tensor (bool) – allow image types that are not tensors (i.e. numpy ndarrays and PIL Images). Default False, which is recommended so that the dataset is asserted to return tensors to be compatible with deepinv.

Examples using check_dataset:#

Bring your own dataset

Bring your own dataset