alignn
alignn copied to clipboard
RuntimeError: The size of tensor a (536) must match the size of tensor b (300) at non-singleton dimension 1
My regression output is 536 values, how can I change the code to fix the mismatch?