@@ -55,7 +55,9 @@ def helper_test_keras_v2(script_mode: bool = False, eager_mode: bool = True):
5555 # v1 training APIs are currently not supported
5656 # in ZCC mode with smdebug 0.9 and AWS TF 2.3.0
5757 tf .compat .v1 .disable_eager_execution ()
58- enable_tb = False if tf .__version__ == "2.0.2" else True
58+
59+ # Performance regression in the _make_histogram fn
60+ enable_tb = False if tf .__version__ == "2.0.2" or is_tf_2_3 () else True
5961 with SagemakerSimulator (enable_tb = enable_tb ) as sim :
6062 model = get_keras_model_v2 ()
6163 (x_train , y_train ), (x_test , y_test ) = get_keras_data ()
@@ -118,7 +120,10 @@ def helper_test_keras_v2_json_config(
118120 # v1 training APIs are currently not supported
119121 # in ZCC mode with smdebug 0.9 and AWS TF 2.3.0
120122 tf .compat .v1 .disable_eager_execution ()
121- enable_tb = False if tf .__version__ == "2.0.2" else True
123+
124+ # Performance regression in the _make_histogram fn
125+ enable_tb = False if tf .__version__ == "2.0.2" or is_tf_2_3 () else True
126+
122127 with SagemakerSimulator (json_file_contents = json_file_contents , enable_tb = enable_tb ) as sim :
123128 model = get_keras_model_v2 ()
124129 (x_train , y_train ), (x_test , y_test ) = get_keras_data ()
0 commit comments