File "train.py", line 315, in main
evaluate(allDataTest, BatchSamplerTest, multiTaskDataLoaderTest, taskParams,
File "/netscratch/sripathi/task2/NLP_Task/multi-task-NLP/models/eval.py", line 25, in evaluate
prediction, scores = model.predict_step(batchMetaData, batchData)
File "/netscratch/sripathi/task2/NLP_Task/multi-task-NLP/models/model.py", line 265, in predict_step
outLogits = self.network(*modelInputs)
File "/netscratch/sripathi/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/netscratch/sripathi/task2/NLP_Task/multi-task-NLP/models/model.py", line 120, in forward
logits = self.allHeaderstaskName
File "/netscratch/sripathi/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/netscratch/sripathi/anaconda3/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 93, in forward
return F.linear(input, self.weight, self.bias)
File "/netscratch/sripathi/anaconda3/lib/python3.8/site-packages/torch/nn/functional.py", line 1688, in linear
if input.dim() == 2 and bias is not None:
AttributeError: 'tuple' object has no attribute 'dim'