Description
Hi everyone,
I encountered an issue while running the following command:
python train.py -opt options/train/train_zsm.yml
I receive the following error:
Debug: Trying to access key: 00081_0565_1_7
Traceback (most recent call last):
File "train.py", line 196, in
main()
File "train.py", line 153, in main
for _, train_data in enumerate(train_loader):
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in next
return self._process_next_batch(batch)
File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
TypeError: function takes exactly 5 arguments (1 given)
It seems like there's an issue with how train_loader is processing the data. The error suggests a mismatch in the number of arguments passed to a function.
I am using Python 3.6 and PyTorch
The dataset is correctly placed, and the configuration file train_zsm.yml is untouched.
Has anyone encountered this error before? Any insights on where this argument mismatch might be occurring?
Any help would be appreciated!
Thanks in advance!