FM / data

/usr/local/lib64/python3.6/site-packages/torch/utils/data/

/usr/local/lib64/python3.6/site-packages/torch/utils/data

Name	Size	Mode	Actions
communication/	-	0755	rm
datapipes/	-	0755	rm
_utils/	-	0755	rm
__pycache__/	-	0755	rm
backward_compatibility.py	254	0644	edit dl rm
dataloader.py	67104	0644	edit dl rm
dataloader_experimental.py	6995	0644	edit dl rm
dataset.py	15001	0644	edit dl rm
distributed.py	5689	0644	edit dl rm
graph.py	1115	0644	edit dl rm
sampler.py	10226	0644	edit dl rm
sharding.py	1079	0644	edit dl rm
_decorator.py	7699	0644	edit dl rm
_typing.py	14436	0644	edit dl rm
__init__.py	2070	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/utils/data/__init__.py (2070B)

__all__ = ['BatchSampler',
           'ChainDataset',
           'ConcatDataset',
           'DataLoader',
           'DataLoader2',
           'Dataset',
           'DistributedSampler',
           'IterDataPipe',
           'IterableDataset',
           'MapDataPipe',
           'RandomSampler',
           'Sampler',
           'SequentialSampler',
           'Subset',
           'SubsetRandomSampler',
           'TensorDataset',
           'WeightedRandomSampler',
           '_DatasetKind',
           'argument_validation',
           'communication',
           'functional_datapipe',
           'get_worker_info',
           'guaranteed_datapipes_determinism',
           'non_deterministic',
           'random_split',
           'runtime_validation',
           'runtime_validation_disabled']

# Please keep this list sorted
assert __all__ == sorted(__all__)

################################################################################
# import subpackage
################################################################################

# TODO(VitalyFedyunin): Rearranging this imports leads to crash, # need to cleanup dependencies and fix it from torch.utils.data.sampler import ( BatchSampler, RandomSampler, Sampler, SequentialSampler, SubsetRandomSampler, WeightedRandomSampler, ) from torch.utils.data.dataset import ( ChainDataset, ConcatDataset, DataChunk, Dataset, Dataset as MapDataPipe, DFIterDataPipe, IterableDataset, IterableDataset as IterDataPipe, Subset, TensorDataset, random_split, ) from torch.utils.data.dataloader import ( DataLoader, _DatasetKind, get_worker_info, ) from torch.utils.data.distributed import DistributedSampler from torch.utils.data._decorator import ( argument_validation, functional_datapipe, guaranteed_datapipes_determinism, non_deterministic, runtime_validation, runtime_validation_disabled, ) from torch.utils.data.dataloader_experimental import DataLoader2 from torch.utils.data import communication __all__ = ['BatchSampler', 'ChainDataset', 'ConcatDataset', 'DataLoader', 'DataLoader2', 'Dataset', 'DistributedSampler', 'IterDataPipe', 'IterableDataset', 'MapDataPipe', 'RandomSampler', 'Sampler', 'SequentialSampler', 'Subset', 'SubsetRandomSampler', 'TensorDataset', 'WeightedRandomSampler', '_DatasetKind', 'argument_validation', 'communication', 'functional_datapipe', 'get_worker_info', 'guaranteed_datapipes_determinism', 'non_deterministic', 'random_split', 'runtime_validation', 'runtime_validation_disabled'] # Please keep this list sorted assert __all__ == sorted(__all__) ################################################################################ # import subpackage ################################################################################