We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 39ac094 commit 112a8fcCopy full SHA for 112a8fc
benchmarks/common/tensorflow/start.sh
@@ -166,6 +166,9 @@ if [[ $NUMA_CORES_PER_INSTANCE != "None" || $SOCKET_ID != "-1" || $NUM_CORES !=
166
elif [[ $MODEL_NAME == "bert_large" && $MODE == "training" && $MPI_NUM_PROCESSES != "None" ]]; then
167
# BERT large training with MPI uses numactl
168
INSTALL_NUMACTL="True"
169
+elif [[ $MODEL_NAME == "wide_deep" ]]; then
170
+ # TODO: Why Wide & Deep uses numactl always
171
+ INSTALL_NUMACTL="True"
172
fi
173
174
# If we are running in a container, call the container_init.sh files
0 commit comments