Training seq2seq translation with multi gpu gives error invalid argument 6: wrong matrix size at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:492

17

Your input should have the shape [seq_len, batch, input_size].
Could you check the current shape? It seems one dimension is missing.

1 Like