class_docstr: Class that defines the methods necessary to hook into process.
class_docstr_tok: ["Class","that","defines","the","methods","necessary","to","hook","into","process","."]
code: def on_train_init(
      self,
      base_config,
      experiment_directory,
      experiment_name,
      model_name,
      output_directory,
      resume_directory,
  ):
      logger.info("wandb.on_train_init() called...")
      wandb.init(
          project=os.getenv("WANDB_PROJECT", experiment_name),
          name=model_name,
          sync_tensorboard=True,
          dir=output_directory,
      )
      wandb.save(os.path.join(experiment_directory, "*"))
  
file_path: ludwig/ludwig/contribs/wandb.py
imports: import logging
  import os
  from ludwig.api_annotations import PublicAPI
  from ludwig.callbacks import Callback
  from ludwig.utils.package_utils import LazyLoader
label_desc: Use image node_id 1 for calling the WandbCallback obj's underlying member method code with example usage: obj.on_train_init(base_config, experiment_directory, experiment_name, model_name, output_directory, resume_directory) without return types