RuntimeError: The size of tensor a (2) must match the size of tensor b (32) at non-singleton dimension 1

[GUZIYIN]

当运行 python main.py --task kamed --super_rate 0.5 --model vrbot --train_batch_size 32 --test_batch_size 32 --device 3 --worker_num 5，出现了RuntimeError: The size of tensor a (2) must match the size of tensor b (32) at non-singleton dimension 1，请问怎么解决？