madeleinegrunde / AGQA_baselines_code

MIT License
18 stars 4 forks source link

HCRN Dataloader #13

Closed danaibrilli closed 1 year ago

danaibrilli commented 1 year ago

Hello! I am trying to run HCRN and followed your instructions for changing the files on the original repo and preprocessing the questions and now I'm trying to run train.py, but I encountered an issue. While running train.py, I get the error:

File "train.py", line 133, in train , , answers, batch_input = [todevice(x, device) for x in batch] ValueError: too many values to unpack (expected 4)

I tried to print the batch's length and it turns out to be 9, as it seems to be at the Dataloader return command as well. Am I missing something? How are only 4 values expected at line 133?

Thank you in advance!

danaibrilli commented 1 year ago

oopsie