diff --git a/data/imagenet10_dataloader.py b/data/imagenet10_dataloader.py index c265120..60bc499 100644 --- a/data/imagenet10_dataloader.py +++ b/data/imagenet10_dataloader.py @@ -1,7 +1,7 @@ import torch +import config import torchvision.datasets as datasets import torchvision.transforms as transforms -import config def get_data_loaders(): print('==> Preparing Imagenet 10 class data..')