Skip to content

Commit f2aab85

Browse files
committed
Disable MKL for TensorFlow iris example to get a better performance.
1 parent 8211a60 commit f2aab85

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

sagemaker-python-sdk/tensorflow_iris_dnn_classifier_using_estimators/iris_dnn_classifier.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44

55
INPUT_TENSOR_NAME = 'inputs'
66

7+
# Disable MKL to get a better perfomance for this model.
8+
os.environ['TF_DISABLE_MKL'] = '1'
9+
os.environ['TF_DISABLE_POOL_ALLOCATOR'] = '1'
10+
711

812
def estimator_fn(run_config, params):
913
feature_columns = [tf.feature_column.numeric_column(INPUT_TENSOR_NAME, shape=[4])]

0 commit comments

Comments
 (0)