archai.datasets package¶
Subpackages¶
- archai.datasets.providers package
- Submodules
- archai.datasets.providers.cifar100_provider module
- archai.datasets.providers.cifar10_provider module
- archai.datasets.providers.fashion_mnist_provider module
- archai.datasets.providers.flower102_provider module
- archai.datasets.providers.food101_provider module
- archai.datasets.providers.imagenet_folder module
- archai.datasets.providers.imagenet_provider module
- archai.datasets.providers.mit67_provider module
- archai.datasets.providers.mnist_provider module
- archai.datasets.providers.sport8_provider module
- archai.datasets.providers.svhn_provider module
- Module contents
- archai.datasets.transforms package
Submodules¶
archai.datasets.aug_policies module¶
archai.datasets.augmentation module¶
archai.datasets.data module¶
-
class
archai.datasets.data.
SubsetSampler
(indices)[source]¶ Bases:
torch.utils.data.sampler.Sampler
Samples elements from a given list of indices, without replacement.
- Arguments:
indices (sequence): a sequence of indices
-
archai.datasets.data.
create_dataset_provider
(conf_dataset: archai.common.config.Config) → archai.datasets.dataset_provider.DatasetProvider[source]¶
-
archai.datasets.data.
get_data
(conf_loader: archai.common.config.Config) → Tuple[Optional[torch.utils.data.dataloader.DataLoader], Optional[torch.utils.data.dataloader.DataLoader], Optional[torch.utils.data.dataloader.DataLoader]][source]¶
-
archai.datasets.data.
get_dataloaders
(ds_provider: archai.datasets.dataset_provider.DatasetProvider, load_train: bool, train_batch_size: int, load_test: bool, test_batch_size: int, aug, cutout: int, val_ratio: float, apex: archai.common.apex_utils.ApexUtils, val_fold=0, train_workers: Optional[int] = None, test_workers: Optional[int] = None, target_lb=- 1, max_batches: int = - 1) → Tuple[Optional[torch.utils.data.dataloader.DataLoader], Optional[torch.utils.data.dataloader.DataLoader], Optional[torch.utils.data.dataloader.DataLoader]][source]¶
archai.datasets.dataset_provider module¶
-
class
archai.datasets.dataset_provider.
DatasetProvider
(conf_dataset: archai.common.config.Config)[source]¶ Bases:
overrides.enforce.EnforceOverrides
archai.datasets.distributed_stratified_sampler module¶
-
class
archai.datasets.distributed_stratified_sampler.
DistributedStratifiedSampler
(dataset: torch.utils.data.dataset.Dataset, world_size: Optional[int] = None, rank: Optional[int] = None, shuffle=True, val_ratio: Optional[float] = 0.0, is_val=False, max_items: Optional[int] = None)[source]¶ Bases:
torch.utils.data.sampler.Sampler