StopIteration error when running code

[DivyanshuSheth]

Hi, I'm encountering the below error while running the code for few-shot training:

***** Running training ***** Num Iters: 4 Batch size: 8 Num steps: 80 Epoch: 0%| | 0/20 [00:12<?, ?it/s] Traceback (most recent call last): File "train_task.py", line 381, in main() File "train_task.py", line 315, in main loss, score = ForwardModelsTrain(config, task_cfg, device, task, batch, model, criterion) File "/home/dasheth/multimodal/iglue/volta/volta/task_utils.py", line 248, in ForwardModelsTrain segment_ids, input_mask, image_mask) File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 171, in forward outputs = self.parallel_apply(replicas, inputs, kwargs) File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 181, in parallel_apply return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 89, in parallel_apply output.reraise() File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/_utils.py", line 543, in reraise raise exception StopIteration: Caught StopIteration in replica 0 on device 0. Original Traceback (most recent call last): File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 64, in _worker output = module(*input, **kwargs) File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/dasheth/multimodal/iglue/volta/volta/encoders.py", line 1223, in forward output_all_attention_masks=output_all_attention_masks, File "/home/dasheth/miniconda3/envs/multimodal_iglue/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "/home/dasheth/multimodal/iglue/volta/volta/encoders.py", line 988, in forward dtype=next(self.parameters()).dtype StopIteration

What could be the issue? I appreciate your help, thank you!