sc2_datasets.torch.datasets.sc2_dataset_directory

Classes

Module Contents

class SC2DatasetDirectory(directory: pathlib.Path, transform: None | Callable = None, validator: None | Callable = None)

Bases: torch.utils.data.Dataset

directory
transform = None
validator = None
skip_files: set[pathlib.Path]
list_of_files: list[pathlib.Path] = []
len = 0
__len__() int
__getitem__(index: int)