code: def on_epoch_end(self, epoch_number: int, logs: Dict) -> None:
      self._on_epoch_end_log(epoch_number, logs)
  
imports: from typing import Dict, Iterable, List, Tuple
  import numpy
  import torch
  from poutyne.framework.callbacks.callbacks import Callback
label_desc: Use image node_id 4 for calling the GradientTracker obj's underlying member method code with example usage: obj.on_epoch_end(epoch_number, logs) without return types