From f581ab6d5c5b83a0625db3552d741096864b61c4 Mon Sep 17 00:00:00 2001 From: Rean <125299691+n0cct1s@users.noreply.github.com> Date: Mon, 20 Jan 2025 23:35:00 -0500 Subject: [PATCH] Update imagenet10_dataloader.py --- data/imagenet10_dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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..')