Shipping metrics to a third party tool is fairly straightforward. In this example, we’ll log our training metrics to Weights and Biases (WandB) to track our experiments.
os.environ
object. We’ll login to WandB with our API key:
wandb.log()
method will record whatever data you wish to pass in during your training process.