-
Notifications
You must be signed in to change notification settings - Fork 2
Weights & Biases
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 ...
for _ in range(num_epochs): train_model() loss = calulate_loss() wandb.log({"Loss": loss})
np.save("weights", weights) wandb.save("weights.npy")`
Execute training script as normal. Additional wandb logging entries will be displayed.