Weights & Biases - mcdomx/cs109b_groupproject GitHub Wiki

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 from wandb.keras import WandbCallback run = wandb.init(project="cs109b_groupproject") run.<model parameter> = XX (e.g. run.batch_size = 64) ... model instantiation code ... # Log metrics with wandb model.fit(X_train, y_train, validationData=(X_test, y_test) epochs=config.epochs, callbacks=[WandbCallback()]) # Save model to wandb model.save(os.path.join(wandb.run.dir, "model.h5"))