I run this line right before the training loop:
wandb.config = {
"train_steps": train_steps,
"batch_size": batch_size,
"unet1_dim": unet1_dim,
"unet2_dim": unet2_dim,
"unet_training": unet_training,
}
But on the dashboard, I can’t seem to find anything related to batch size. All of the data logged with wandb.log
is present, but nothing about my hyperparameters. It may be relevant that I’m running offline and syncing with the command wandb sync --sync-all
.