code: def on_epoch_begin(self, epoch_number: int, logs: Dict):
      self.epoch = epoch_number
  
imports: import csv
  import os
  from typing import Dict, List, Optional, TextIO, Union
  import torch
  from poutyne.framework.callbacks._utils import atomic_lambda_save
  from poutyne.framework.callbacks.callbacks import Callback
label_desc: Use image node_id 3 for calling the GradientLoggerBase obj's underlying member method code with example usage: obj.on_epoch_begin(epoch_number, logs) without return types