semi-siamese training error

[PatWolf2610]

An error happen while I try to train the ResNet model with my custom dataset on GoogleColab
/content/FaceX-Zoo/training_mode/semi-siamese_training/train_vg.sh: 3: /content/FaceX-Zoo/training_mode/semi-siamese_training/train_vg.sh: mkdire: not found
INFO 2022-10-20 04:51:36 train.py: 189] Start optimization.
INFO 2022-10-20 04:51:36 train.py: 190] Namespace(alpha=0.999, backbone_conf_file='/content/FaceX-Zoo/training_mode/backbone_conf.yaml', backbone_type='ResNet', batch_size=8, data_root='/content/VN_celeb', epoches=250, evaluate=False, head_conf_file='/content/FaceX-Zoo/training_mode/head_conf.yaml', head_type='ArcFace', log_dir='log', lr=0.1, milestones=[60, 100, 120], momentum=0.9, out_dir='log/model', pretrain_model='/content/resnet152.pt', print_freq=20, resume=False, save_freq=10, step='60,100,120', tensorboardx_logdir='log/board', test_data_conf_file=None, test_set='LFW', train_file='/content/FaceX-Zoo/training_mode/train_list.txt', writer=<tensorboardX.writer.SummaryWriter object at 0x7fe18ac7ba50>)
backbone param:
{'depth': 152, 'drop_ratio': 0.4, 'net_mode': 'ir_se', 'feat_dim': 512, 'out_h': 7, 'out_w': 7}
head param:
{'feat_dim': 512, 'num_class': 72778, 'margin_arc': 0.35, 'margin_am': 0.0, 'scale': 32}
Valid ids: 1020.
/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py:566: UserWarning: This DataLoader will create 4 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
cpuset_checked))
Traceback (most recent call last):
File "/content/FaceX-Zoo/training_mode/semi-siamese_training/train.py", line 191, in
train(args)
File "/content/FaceX-Zoo/training_mode/semi-siamese_training/train.py", line 125, in train
prototype, optimizer, criterion, epoch, conf, loss_meter)
File "/content/FaceX-Zoo/training_mode/semi-siamese_training/train.py", line 77, in train_one_epoch
images1_probe, images2_gallery, images2_probe, images1_gallery, id_indexes)
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
TypeError: forward() takes 3 positional arguments but 6 were given