Multiple gpu errors

[yssjglg-elder]

RuntimeError: Sizes of tensors must match except in dimension 0. Got 32 and 16 (The offending index is 0)

This error will appear in ‘Validation sanity check‘

If I use multiple GPU training, the above error will appear. I think it's a bug in generating the data rather than a dimension mismatch, but I can't fix it, do you have any idea about that？