Skip to content

Commit 6358a84

Browse files
author
David Goodwin
committed
Remove Caffe2 from resnet bs=128 perf testing for now
The same OOM problem impacting pytorch hits caffe2 as well, likely because they share an execution engine.
1 parent 6ab5947 commit 6358a84

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

qa/L0_perf_resnet/test.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ done
123123
STATIC_BATCH=128
124124
DYNAMIC_BATCH=1
125125
INSTANCE_CNT=2
126-
MODEL_NAMES="${TRT_MODEL_NAME} ${TF_MODEL_NAME} ${NETDEF_MODEL_NAME}"
126+
MODEL_NAMES="${TRT_MODEL_NAME} ${TF_MODEL_NAME}"
127127
OPTIMIZED_MODEL_NAMES="${TFTRT_MODEL_NAME}"
128128

129129
# Create the TensorRT plan from Caffe model

0 commit comments

Comments
 (0)