Dataset Collection MNIST In config.yaml file: setting dataset_mode: MNIST CIFAR10 In config.yaml file: setting dataset_mode: CIFAR10