From 26ca22b2fd82c8528debccb265a0a4aae32f7244 Mon Sep 17 00:00:00 2001 From: Hust_XianYang Date: Tue, 22 Oct 2019 11:17:10 +0800 Subject: [PATCH] Update task_generator.py solve the error: "IndexError: too many indices for tensor of dimension 0" by chaging the normalize to one channel --- omniglot/task_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/omniglot/task_generator.py b/omniglot/task_generator.py index f0945d0d..82b89886 100644 --- a/omniglot/task_generator.py +++ b/omniglot/task_generator.py @@ -138,8 +138,8 @@ def __len__(self): def get_data_loader(task, num_per_class=1, split='train',shuffle=True,rotation=0): # NOTE: batch size here is # instances PER CLASS - normalize = transforms.Normalize(mean=[0.92206, 0.92206, 0.92206], std=[0.08426, 0.08426, 0.08426]) - + # normalize = transforms.Normalize(mean=[0.92206, 0.92206, 0.92206], std=[0.08426, 0.08426, 0.08426]) + normalize=transforms.Normalize(mean=[0.92206],std=[0.8426]) dataset = Omniglot(task,split=split,transform=transforms.Compose([Rotate(rotation),transforms.ToTensor(),normalize])) if split == 'train':