power_cogs.dataset package

Submodules

power_cogs.dataset.mnist_dataset module

class power_cogs.dataset.mnist_dataset.MNISTDataset[source]

Bases: power_cogs.base.base_torch_dataset.BaseTorchDataset

to_device(device: torch.device)[source]

Module contents