From 6defd240007d1f5361cc3ad67edc0ec7df494d06 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 15 Sep 2022 19:56:17 +0100 Subject: [PATCH] bugfix: too many values to unpack --- aimodel/src/lib/dataset/batched_iterator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/batched_iterator.py b/aimodel/src/lib/dataset/batched_iterator.py index 9f12d73..3c45dc2 100644 --- a/aimodel/src/lib/dataset/batched_iterator.py +++ b/aimodel/src/lib/dataset/batched_iterator.py @@ -10,7 +10,8 @@ def batched_iterator(dataset, tensors_in_item=1, batch_size=64): if tensors_in_item == 1: item = [ item ] - for i_tensor, tensor in item: + + for i_tensor, tensor in enumerate(item): acc[i_tensor].append(tensor) if i_item >= batch_size: