class_docstr: A stripped-down callback which is focused on batches rather than epochs.
docstr: Hook called after every training batch.
docstr_tok: ["Hook","called","after","every","training","batch","."]
code: def on_train_batch_end(
      self, network: nn.HybridBlock, time_elapsed: float
  ) -> None:
      """
      Hook called after every training batch.
      """
  
file_path: gluonts/src/gluonts/nursery/tsbench/src/tsbench/gluonts/callbacks/base.py
imports: from typing import List
  from mxnet import gluon
  from mxnet.gluon import nn
label_desc: Use image node_id 3 for calling the Callback obj's underlying member method code with example usage: obj.on_train_batch_end(network, time_elapsed) without return types