The actual error is:
RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 3 and 4 at /opt/conda/conda-bld/pytorch_1524586445097/work/aten/src/TH/generic/THTensorMath.c:3577

It seems your dataset is returning data of different dimensions and the dataloader couldn’t handle it.