parameters: (self, step, logs=None)
docstring: Called at the end of each batch in training.
  
          Args:
              step (int): The index of step (or iteration).
              logs (dict): The logs is a dict or None. The `logs` passed by
                  paddle.Model is a dict, contains 'loss', metrics and 'batch_size'
                  of current batch.
docstring_summary: Called at the end of each batch in training.
docstring_tokens: Called at the end of each batch in training .
function: def on_train_batch_end(self, step, logs=None):
          """Called at the end of each batch in training.
  
          Args:
              step (int): The index of step (or iteration).
              logs (dict): The logs is a dict or None. The `logs` passed by
                  paddle.Model is a dict, contains 'loss', metrics and 'batch_size'
                  of current batch.
          """
function_tokens: def on_train_batch_end ( self , step , logs = None ) :