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