RuntimeError: invalid argument 2: size '[2 x 5 x 5 x 128]' is invalid for input of with 12800 elements at /opt/conda/conda-bld/pytorch_1511304568725/work/torch/lib/TH/THStorage.c:41

[abhimanyuchadha96]

Traceback (most recent call last):
File "levers.py", line 292, in
game.infer()
File "levers.py", line 281, in infer
action_prob, _baseline = self.communicate(agent_ids, passes=2)
File "levers.py", line 150, in communicate
self.comm_in = self.comm_in.view(self.batch_size, self.nagents, self.nagents, self.hidsz)
File "/home/abhimanyu/anaconda3/envs/Autonomous_Project/lib/python2.7/site-packages/torch/autograd/variable.py", line 510, in view
return View.apply(self, sizes)
File "/home/abhimanyu/anaconda3/envs/Autonomous_Project/lib/python2.7/site-packages/torch/autograd/_functions/tensor.py", line 96, in forward
result = i.view(*sizes)
RuntimeError: invalid argument 2: size '[2 x 5 x 5 x 128]' is invalid for input of with 12800 elements at /opt/conda/conda-bld/pytorch_1511304568725/work/torch/lib/TH/THStorage.c:41