Model config parameters not saved with wandb sync

W&B Help
1

Hello,

I’m using wandb with pytorch-lightning, specifically with the WandbLogger.

The issue I’m encountering is that the config parameters do not appear to be synced to the dashboard, even though I can find the configuration in the wandb log files:

2024-12-09 11:49:59,236 INFO    MainThread:75836 [wandb_run.py:_config_callback():1387] config_cb None None {'inpt_dim': "{'cond': [1], 'data': [6]}", 'discrete_indices': '[0, 1]', 'discrete_shift': 5, 'discrete_scale': 3, 'num_blocks': 8, 'num_layers': 1, 'num_hidden': 128, 'num_bins': 8, 'tail_bound': 6, 'batch_norm': False, 'lr': 0.0001, 'weight_decay': 1e-05}
2024-12-09 12:31:39,615 WARNING MsgRouterThr:75836 [router.py:message_loop():75] message_loop has been closed

image
image1578×416 33.2 KB

Any advice would be greatly appreciated.

Cheers,
Vilius

2

I can also reproduce the same issue when running this notebook: Google Colab

after making the following change:

wandb_logger = WandbLogger(project="lit-wandb", offline=True)