train.dataset package

Submodules

train.dataset.cifar100_dataloader module

Create train, valid, main iterators for CIFAR-10 [1]. Easily extended to MNIST, CIFAR-100 and Imagenet. [1]: https://discuss.pytorch.org/t/feedback-on-pytorch-for-kaggle-competitions/2252/4

class train.dataset.CIFAR100.CIFAR100

Bases: train.dataset.dataloader.BaseDataloader

get_test_dataloader()

get_train_dataloader()

get_val_dataloader()

train.dataset.cifar10_dataloader module

Create train, valid, main iterators for CIFAR-10 [1]. Easily extended to MNIST, CIFAR-100 and Imagenet. [1]: https://discuss.pytorch.org/t/feedback-on-pytorch-for-kaggle-competitions/2252/4

class train.dataset.CIFAR10.CIFAR10

Bases: train.dataset.dataloader.BaseDataloader

get_test_dataloader()

get_train_dataloader()

get_val_dataloader()

train.dataset.dataloader module

class train.dataset.dataloader.BaseDataloader

Bases: object

Abstract class for Dataloader

get_test_dataloader()

get_train_dataloader()

get_val_dataloader()

Module contents