Skip to content

Weights & Biases

Mark McDonald edited this page Jan 28, 2020 · 13 revisions

Weights & Biases is an excellent tool to support training by running a series of alternate parameters on your models and logging the results. Wandb makes it easy to fine the best parameters and recall them to train your final model with.

The project page is at https://app.wandb.ai/mcdomx/cs109b_groupproject

Install and Setup: pip install --upgrade wandb wandb login <wandb token> (visit project page and login)

Include WANDB in Training Script: `import wandb wandb.init(project="cs109b_groupproject") Model instantiation code ...

Log metrics with wandb

for _ in range(num_epochs): train_model() loss = calulate_loss() wandb.log({"Loss": loss})

Save model to wandb

np.save("weights", weights) wandb.save("weights.npy")`

Execute training script as normal. Additional wandb logging entries will be displayed.

Clone this wiki locally