I did double-check and the error message is indeed the same with the same tensor sizes and thrown at the same line.  The code is updated. Here is the error thrown
size mismatch, m1: [1 x 20592], m2: [20526 x 20658] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:290