It is working to some extent but when the data splitted as tensor is given to dataloader shows errors
RuntimeError: size mismatch, m1: [1 x 1], m2: [921 x 100] at /opt/conda/conda-bld/pytorch_1532571933727/work/aten/src/THC/generic/THCTensorMathBlas.cu:249
Could you pls help me to identify the problem?