code: def on_train_start(self, tez_trainer, **kwargs):
      self._train_tqdm = tqdm(
          total=self.num_train_steps,
          disable=not tez_trainer._driver.is_main_process,
      )
      if self.num_valid_steps:
          self._valid_tqdm = tqdm(
              total=self.num_valid_steps,
              disable=not tez_trainer._driver.is_main_process,
          )
  
file_path: tez/tez/callbacks/progress.py
imports: from tqdm.auto import tqdm
  from tez.callbacks import Callback
label_desc: Use image node_id 2 for calling the Progress obj's underlying member method code with example usage: obj.on_train_start(tez_trainer) without return types