Weights & Biases

Weights & Biases, a popular experiment tracking tool, reads its API Key from an environment variable named WANDB_API_KEY.

To enable this within MosaicML platform, create an environment variable secret as:

mcli create secret env --name wandb WANDB_API_KEY=<your-wandb-api-key>

Then, test the API key was added correctly with a small test wandb login call:

name: wandb-login
compute:
  gpus: 0
image: python
integrations:
  - integration_type: pip_packages
    packages:
      - wandb
command: |
  wandb login

If secret registration was successful and your API key works, then you should see an output like the following on the command line of your test run’s log:

wandb: Currently logged in as: <wandb user name>