diff --git a/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh b/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh index eb9b1d165d02246be92c21d946dd39aaa8c377a2..a808f35aef441e49f4e57099d91be7f6d9d0e5d1 100644 --- a/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh +++ b/PyTorch/dev/cv/image_classification/ADLayer_ID1087_for_PyTorch/test/train_performance_1p.sh @@ -68,6 +68,7 @@ nohup python3 main.py --dataset_name OpenImages \ --workers 4 \ --data_root $data_path \ --mask_root $data_path \ + --metadata_root $data_path/metadata/ \ --box_v2_metric True \ --iou_threshold_list 30 50 70 \ --eval_checkpoint_type last > $cur_path/test/output/$ASCEND_DEVICE_ID/train_$ASCEND_DEVICE_ID.log 2>&1 &