RuntimeError                              Traceback (most recent call last)
<ipython-input-15-885795271e1c> in <module>()
    111         num_fine=num_fine, mode="train", lindisp=False, perturb=True,
    112         encode_position_fn=encode_position_fn,
--> 113         encode_direction_fn=encode_direction_fn
    114     )
    115 

7 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in linear(input, weight, bias)
   1688     if input.dim() == 2 and bias is not None:
   1689         # fused op is marginally faster
-> 1690         ret = torch.addmm(bias, input, weight.t())
   1691     else:
   1692         output = input.matmul(weight.t())

RuntimeError: mat1 dim 1 must match mat2 dim 0