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 8211a60 commit f2aab85Copy full SHA for f2aab85
sagemaker-python-sdk/tensorflow_iris_dnn_classifier_using_estimators/iris_dnn_classifier.py
@@ -4,6 +4,10 @@
4
5
INPUT_TENSOR_NAME = 'inputs'
6
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
+
11
12
def estimator_fn(run_config, params):
13
feature_columns = [tf.feature_column.numeric_column(INPUT_TENSOR_NAME, shape=[4])]
0 commit comments