mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
39116409a1
Including: - `torch/utils/benchmarks/` - `torch/utils/data/` Fixes part of #164878 Pull Request resolved: https://github.com/pytorch/pytorch/pull/165299 Approved by: https://github.com/albanD
79 lines
1.7 KiB
Python
79 lines
1.7 KiB
Python
from torch.utils.data.dataloader import (
|
|
_DatasetKind,
|
|
DataLoader,
|
|
default_collate,
|
|
default_convert,
|
|
get_worker_info,
|
|
)
|
|
from torch.utils.data.datapipes._decorator import (
|
|
argument_validation,
|
|
functional_datapipe,
|
|
guaranteed_datapipes_determinism,
|
|
non_deterministic,
|
|
runtime_validation,
|
|
runtime_validation_disabled,
|
|
)
|
|
from torch.utils.data.datapipes.datapipe import (
|
|
DataChunk,
|
|
DFIterDataPipe,
|
|
IterDataPipe,
|
|
MapDataPipe,
|
|
)
|
|
from torch.utils.data.dataset import (
|
|
ChainDataset,
|
|
ConcatDataset,
|
|
Dataset,
|
|
IterableDataset,
|
|
random_split,
|
|
StackDataset,
|
|
Subset,
|
|
TensorDataset,
|
|
)
|
|
from torch.utils.data.distributed import DistributedSampler
|
|
from torch.utils.data.sampler import (
|
|
BatchSampler,
|
|
RandomSampler,
|
|
Sampler,
|
|
SequentialSampler,
|
|
SubsetRandomSampler,
|
|
WeightedRandomSampler,
|
|
)
|
|
|
|
|
|
__all__ = [
|
|
"BatchSampler",
|
|
"ChainDataset",
|
|
"ConcatDataset",
|
|
"DFIterDataPipe",
|
|
"DataChunk",
|
|
"DataLoader",
|
|
"Dataset",
|
|
"DistributedSampler",
|
|
"IterDataPipe",
|
|
"IterableDataset",
|
|
"MapDataPipe",
|
|
"RandomSampler",
|
|
"Sampler",
|
|
"SequentialSampler",
|
|
"StackDataset",
|
|
"Subset",
|
|
"SubsetRandomSampler",
|
|
"TensorDataset",
|
|
"WeightedRandomSampler",
|
|
"_DatasetKind",
|
|
"argument_validation",
|
|
"default_collate",
|
|
"default_convert",
|
|
"functional_datapipe",
|
|
"get_worker_info",
|
|
"guaranteed_datapipes_determinism",
|
|
"non_deterministic",
|
|
"random_split",
|
|
"runtime_validation",
|
|
"runtime_validation_disabled",
|
|
]
|
|
|
|
# Please keep this list sorted
|
|
if __all__ != sorted(__all__):
|
|
raise AssertionError("__all__ is not sorted")
|