Skip to content

Commit

Permalink
link fix (activeloopai#1558)
Browse files Browse the repository at this point in the history
  • Loading branch information
AbhinavTuli authored Mar 23, 2022
1 parent c1d21e0 commit 4a49798
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion hub/core/dataset/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -888,7 +888,7 @@ def pytorch(
Read torch.utils.data.DataLoader docs for more details.
pin_memory (bool): If True, the data loader will copy Tensors into CUDA pinned memory before returning them. Default value is False.
Read torch.utils.data.DataLoader docs for more details.
shuffle (bool): If True, the data loader will shuffle the data indices. Default value is False. Details about how hub shuffles data can be found at https://docs.activeloop.ai/how-hub-works/shuffling-in-ds.pytorch.
shuffle (bool): If True, the data loader will shuffle the data indices. Default value is False. Details about how hub shuffles data can be found at https://docs.activeloop.ai/how-hub-works/shuffling-in-ds.pytorch
buffer_size (int): The size of the buffer used to shuffle the data in MBs. Defaults to 2048 MB. Increasing the buffer_size will increase the extent of shuffling.
use_local_cache (bool): If True, the data loader will use a local cache to store data. This is useful when the dataset can fit on the machine and we don't want to fetch the data multiple times for each iteration. Default value is False.
use_progress_bar (bool): If True, tqdm will be wrapped around the returned dataloader. Default value is True.
Expand Down

0 comments on commit 4a49798

Please sign in to comment.