From a2f389015866f443ab544595b253e919ab7a97fe Mon Sep 17 00:00:00 2001 From: NihalHarish Date: Sun, 2 Aug 2020 02:47:28 -0700 Subject: [PATCH] rename enum --- smdebug/tensorflow/keras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/smdebug/tensorflow/keras.py b/smdebug/tensorflow/keras.py index d34b645be..ccea3ba47 100644 --- a/smdebug/tensorflow/keras.py +++ b/smdebug/tensorflow/keras.py @@ -454,7 +454,7 @@ def save_smdebug_logs(self, logs): elif key == SMDEBUG_LAYER_OUTPUTS_KEY: layer_outputs = logs[key] self.save_layer_outputs(layer_outputs) - self.save_layer_inputs(logs[ModelInput.X], layer_outputs) + self.save_layer_inputs(logs[ModelInput.INPUTS], layer_outputs) # Save Model Inputs elif key in ModelInputs: export_name = get_model_input_export_name()