Yuan Hao

Problem: Index tensor must have same dimensions as output tensor

4

Traceback (most recent call last): File "train.py", line 154, in loss.backward() File "/usr/local/lib/python3.5/dist-packages/torch/autograd/variable.py", line 152, in backward torch.autograd.backward(self, gradient, retain_graph, create_graph, retain_variables) File "/usr/local/lib/python3.5/dist-packages/torch/autograd/__init__.py", line 98, in backward variables, grad_variables,...