Hi,
I am getting this error
RuntimeError: The shape of the mask [8, 32756] at index 0 does not match the shape of the indexed tensor [262048, 1] at index 0
Exception ignored in: <bound method _DataLoaderIter.del of <torch.utils.data.dataloader._DataLoaderIter object at 0x7f801b6ccf28>>'
I am using pytorch 0.4.0. and python 3.5 anaconda 4.5.9.
Any help? Thanks