Comment the checking operation in  site-packages/torch/utils/data/_utils/collate.py (line 80 81)
#if not all(len(elem) == elem_size for elem in it):
        #    raise RuntimeError('each element in list of batch should be of equal size')

That works for me with torch1.8 on Windows10.