Thank you… the problem is solved! but I have encounter other problem:
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 385 and 341 in dimension 2 at /opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/TH/generic/THTensorMath.c:3586
I’m not sure if this related to the batch size I tried to manipulate it but still the error exists, any tips please?