class_docstr: Callback that streams epoch results to a [Weights & Biases](https://self.wandb.ai/) run.
  
  ```python
  self.wandb.login()
  wandb_logger = WandbCallback(project="sample-self.wandb-project", job_type="train")
  model.fit(X_train, Y_train, callbacks=[wandb_logger])
  ```
class_docstr_tok: ["Callback","that","streams","epoch","results","to","a","[","Weights","&","Biases","]","(","https",":","\/\/self.wandb.ai\/",")","run",".","``","`","python","self.wandb.login","(",")","wandb_logger","=","WandbCallback","(","project=","''","sample-self.wandb-project","''",",","job_type=","''","train","''",")","model.fit","(","X_train",",","Y_train",",","callbacks=","[","wandb_logger","]",")","``","`"]
code: def on_train_begin(self, logs=None):
      self.steps = self.params["steps"]
      self.global_step = 0
      self._gather_configs()
  
file_path: elegy/elegy/callbacks/wandb_callback.py
imports: import math
  from typing import Dict, Optional, Union
  from .callback import Callback
label_desc: Use image node_id 4 for calling the WandbCallback obj's underlying member method code with example usage: obj.on_train_begin(logs) without return types