From f8bfd3493af8881cc2e1c4402a516ecfd26c0e55 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Thu, 2 Jun 2022 13:54:57 -0700 Subject: [PATCH] make destructuring datum length agnostic when validating in training decoder script, for @YUHANG-Ma --- train_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_decoder.py b/train_decoder.py index 057ca6d..76b49c2 100644 --- a/train_decoder.py +++ b/train_decoder.py @@ -331,7 +331,7 @@ def train( sample = 0 average_loss = 0 timer = Timer() - for i, (img, emb, txt) in enumerate(dataloaders["val"]): + for i, (img, emb, *_) in enumerate(dataloaders["val"]): sample += img.shape[0] img, emb = send_to_device((img, emb))