microsoft / Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2
Other
1.9k stars 345 forks source link

[wandb] disable wandb more gracefully #422

Closed billishyahao closed 3 months ago

billishyahao commented 4 months ago

Currently we can only disable wandb by removing its package (pip uninstall wandb). It is not graceful if users have wandb installed in their dev environment but dont want to use wandb. With this patch, wandb feature can be disabled if both exp or project name are not specified.

loadams commented 4 months ago

Thanks for this change @billishyahao - we will take a look at this and review soon.