remove print statement

Author: sophiayue1116

smdebug/tensorflow/keras.py

@@ -1104,7 +1104,7 @@ def unwrap(func):
 
     def close(self):
         self._cleanup()
-
+        # print("\nStep Number in the close function: ", self.step)
         if self.python_profiler:
             # print("python profiling for end of last train step to end of training")
             self.python_profiler.start_profiling(
@@ -1152,6 +1152,8 @@ def run(*args, **kwargs):
                 self.prepared_collections = True
 
             self._increment_step()
+            # print("\nStep number in the push tape: ", self.step)
+
 
             if self._get_collections_to_save_for_step():
                 # print('\n Collections saved for this step: ', self._get_collections_to_save_for_step())
@@ -1242,6 +1244,7 @@ def run(*args, **kwargs):
                 return
 
             self.last_saved_step = self.step
+            # print("\nStep number in the pop tape: ", self.step)
 
         return run
 
@@ -1318,7 +1321,7 @@ def profiling_start_batch(self, mode):
             if self.mode != ModeKeys.GLOBAL:
                 self.mode_steps[ModeKeys.GLOBAL] = self.step
 
-        print("Step Number in start train batch: ", self.mode_steps[mode])
+        # print("Step Number in start train batch: ", self.mode_steps[mode])
 
         self.profiler_config_parser.load_config()
 
@@ -1380,6 +1383,7 @@ def profiling_end_batch(self, mode):
         """
         Enabling profiler at the end of train batch when native tf2 training is used.
         """
+        # print("Step Number in end train batch: ", self.mode_steps[mode])
         if self._is_not_supported():
             return
 
@@ -1412,6 +1416,7 @@ def profiling_end(self):
         """
         Stop profiler at the end of training when native tf2 training is used.
         """
+        # print("Step Number at the end of training: ", self.step)
         # Unwrap the tape before closing and close the python profiling
         self.close()