multiml.task.pytorch.datasets.storegate_dataset module
- class multiml.task.pytorch.datasets.storegate_dataset.StoreGateDataset(storegate, phase, device=None, preload=None, input_var_names=None, true_var_names=None, callbacks=None)
Bases:
Dataset
StoreGate dataset class.
- __init__(storegate, phase, device=None, preload=None, input_var_names=None, true_var_names=None, callbacks=None)
- __len__()
- __getitem__(index)