OVERVIEW
Machine learning experiment tracking
Why is experiment tracking so important for doing real world machine learning? Here's a quick overview of the problem and tools.
Experiment tracking for
machine learning models
import torch
import torch.nn as nn
import wandb
wandb.init(project="pedestrian-detection")
# Log any metric from your training script
wandb.log({"acc": accuracy, "val_acc": val_accuracy})