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..85c13702f6b524d69d84e7db941db7adf1bda5d0 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
@@ -66,6 +66,7 @@ nohup python3 main.py --dataset_name OpenImages \
                --weight_decay 5.00E-04 \
                --override_cache True \
                --workers 4 \
+               --middle_output=$data_path/middle_data \
                --data_root $data_path \
                --mask_root $data_path \
                --box_v2_metric True \