rename enum (#305)

awslabs · Aug 2, 2020 · 85a7a76 · 85a7a76
1 parent 8565c6a
commit 85a7a76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()