docs: def _create_data_loader(self, data, **kwargs):
          """Converts input data into a DataLoader"""
          if data is None:
              return None
  
          # Set DataLoader config
          # NOTE: Not applicable if data is already a DataLoader
          config = {
              **self.config["train_config"]["data_loader_config"],
              **kwargs,
              "pin_memory": self.config["device"] != "cpu",
          }
          # Return data as DataLoader
          if isinstance(data, DataLoader):
              return data
          elif isinstance(data, Dataset):
              return DataLoader(data, **config)
          elif isinstance(data, (tuple, list)):
              return DataLoader(self._create_dataset(*data), **config)
          else:
              raise ValueError("Input data type not recognized.")