I try to train the model using code and the code has some errors

[zj56]

python3 main.py \ --model 'LASTED' \ --train_file 'annotation/Test_num1047.txt' \ --num_class 4 \ --val_ratio 0.005 \ --test_file 'annotation/Test_num410.txt' \ --isTrain 1 \ --lr 0.0001 \ --data_size 448 \ --batch_size 48 \ --gpu '0,1,2,3' \ 2>&1 | tee weights/log.log

[HighwayWu]

Thanks for your asking. Accoding to the error, the input should be 4D tensor, which means the [48, 1, 3, 448, 448] => [48, 3, 448, 448].