使用自制数据集训练时出错

我想用 insightface 做一个 牛脸的认证， blazeface 训练成功了；但是；；；第二步；；；/home/aistudio/insightface/recognition/arcface_paddle
!python tools/train.py \
--config_file configs/ms1mv2_mobileface.py \
--embedding_size 128 \
--sample_ratio 1.0 \
--loss ArcFace \
--batch_size 5 \
--dataset MS1M_v2 \
--num_classes 54 \
--data_dir MS1M_v2/ \
--label_file MS1M_v2/label.txt \
--fp16 False；；；报 File "/home/aistudio/insightface/recognition/arcface_paddle/dynamic/train.py", line 200, in train
for step, (img, label) in enumerate(train_loader): raise ValueError('all input arrays must have the same shape')
ValueError: all input arrays must have the same shape；；我已经把图片缩放为统一大小为啥还是会出现这个错；；；和embedding_size有关系吗