docstr: Called at the end of an epoch.
  Subclasses should override for any actions to run. This function should only
  be called during TRAIN mode.
  :param epoch:  Integer, index of epoch.
  :param logs: Dict, metric results for this training epoch, and for the validation epoch if
      validation is performed. Validation result keys are prefixed with val_. For training
      epoch, the values of the Model's metrics are returned.
      Example : {'loss': 0.2, 'accuracy': 0.7}
code: def after_train_epoch(self, runner):
      """
      Called at the end of an epoch.
      Subclasses should override for any actions to run. This function should only
      be called during TRAIN mode.
      :param epoch:  Integer, index of epoch.
      :param logs: Dict, metric results for this training epoch, and for the validation epoch if
          validation is performed. Validation result keys are prefixed with val_. For training
          epoch, the values of the Model's metrics are returned.
          Example : {'loss': 0.2, 'accuracy': 0.7}
      """
      self.run.log(runner.epoch_stats)
  
file_path: BigDL/python/orca/src/bigdl/orca/learn/pytorch/callbacks/wandb.py
imports: from bigdl.orca.learn.pytorch.callbacks import Callback
  from bigdl.dllib.utils.log4Error import invalidInputError
  import copy
label_desc: Use image node_id 2 for calling the WandbLoggerCallback obj's underlying member method code with example usage: obj.after_train_epoch(runner) without return types