Skip to content

Commit f3abdb5

Browse files
CaoEDiweiSun
andauthored
fix yolov7 inductor script (#2447)
Co-authored-by: diwei sun <[email protected]>
1 parent 099b535 commit f3abdb5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models_v2/pytorch/yolov7/inference/cpu/run_model.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,10 +187,10 @@ else
187187
export TORCHINDUCTOR_FREEZING=1
188188
python -m torch.backends.xeon.run_cpu \
189189
--enable-jemalloc \
190+
$MODE_ARGS \
190191
--log_path=${OUTPUT_DIR} \
191192
${MODEL_DIR}/yolov7/yolov7.py \
192193
$ARGS \
193-
$MODE_ARGS \
194194
--inductor \
195195
--batch-size $BATCH_SIZE 2>&1 | tee ${OUTPUT_DIR}/yolov7_${mode}_log_${PRECISION}.log
196196
fi

0 commit comments

Comments
 (0)