| 139 | |
| 140 | |
| 141 | def get_handler(name): |
| 142 | if name.lower() == 'mnist': |
| 143 | return DataHandler1 |
| 144 | elif name.lower() == 'fashionmnist': |
| 145 | return DataHandler1 |
| 146 | elif name.lower() == 'svhn': |
| 147 | return DataHandler2 |
| 148 | elif name.lower() == 'cifar10': |
| 149 | return DataHandler3 |
| 150 | elif name.lower() == 'cifar100': |
| 151 | return DataHandler3 |
| 152 | elif name.lower() == 'gtsrb': |
| 153 | return DataHandler3 |
| 154 | elif name.lower() == 'tinyimagenet': |
| 155 | return DataHandler3 |
| 156 | else: |
| 157 | return DataHandler4 |
| 158 | |
| 159 | |
| 160 | class DataHandler1(Dataset): |