Skip to content

Commit

Permalink
adds missing dataloader args (#2160)
Browse files Browse the repository at this point in the history
  • Loading branch information
AbhinavTuli committed Feb 6, 2023
1 parent dacc0e4 commit 529b962
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions deeplake/enterprise/dataloader.py
Expand Up @@ -108,8 +108,10 @@ def __init__(

# torch.utils.data.DataLoader attributes
self.__initialized = True
self._IterableDataset_len_called = None
self._iterator = None


@property
def batch_size(self):
return self._batch_size or 1
Expand All @@ -130,14 +132,51 @@ def prefetch_factor(self):
def pin_memory(self):
return False

@property
def pin_memory_device(self):
return ""

@property
def timeout(self):
return 0

@property
def worker_init_fn(self):
return None

@property
def multiprocessing_context(self):
return None

@property
def _dataset_kind(self):
return 1

@property
def sampler(self):
return _InfiniteConstantSampler()

@property
def batch_sampler(self):
return None

@property
def generator(self):
return None

@property
def persistent_workers(self):
return self._persistent_workers or False

@property
def _auto_collation(self):
return False

@property
def _index_sampler(self):
return self.sampler


@property
def collate_fn(self):
return get_collate_fn(self._collate, self._mode)
Expand Down

0 comments on commit 529b962

Please sign in to comment.