The expanded size of the tensor (8) must match the existing size (9) at non-singleton dimension 3. Target sizes: [2, 8, 8, 8]. Tensor sizes: [2, 1, 9, 9]

help me plz ~! i was trying to train swintransformer/OCR but i I got an error

in train dataset image folder has 10 of .“png” photos and 8 of train.txt and 2 of val.txt

"The expanded size of the tensor (8) must match the existing size (9) at non-singleton dimension 3.