RuntimeError: invalid argument 0

[12-wu]

Hello, I have been reading this paper of yours recently, and I am very interested in the semi-supervised classification method you proposed. However, I have encountered some problems when I reproduce this code, so I would like to ask you for some advice. when I run latent-mixing.py, I encountered the following problems:
Traceback (most recent call last):
File "E:/图像分类/2/LatentMixing-master/latent-mixing.py", line 634, in
main()
File "E:/图像分类/2/LatentMixing-master/latent-mixing.py", line 335, in main
train_criterion, epoch, use_cuda, args.mixup, args.noSharp)
File "E:/图像分类/2/LatentMixing-master/latent-mixing.py", line 394, in train
(inputs_x, targets_x) = labeled_train_iter.next()
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 326 and 371 in dimension 3 at C:\w\1\s\windows\pytorch\aten\src\TH/generic/THTensor.cpp:689
Looking forward to your reply!
Thank you very much!