@error_handling_agent.catch_smdebug_errors()
def forward_hook(self, block, inputs, outputs):
    if (not self._get_collections_to_save_for_step()):
        return
    block_name = block.name
    self._write_inputs(block_name, inputs)
    self._write_outputs(block_name, outputs)
    self.last_saved_step = self.step
@error_handling_agent.catch_smdebug_errors()
def forward_hook(self, block, inputs, outputs):
    if (not self._get_collections_to_save_for_step()):
        return
    block_name = block.name
    self._write_inputs(block_name, inputs)
    self._write_outputs(block_name, outputs)
    self.last_saved_step = self.step