python train.py optimizer.lr=0.001

wandb sweep sweep_sgd.yaml

if you run the above command, you will get a sweep id. You can use this id to run the sweep agent.

wandb agent --count 1 <sweep_id>

You can specify the number of runs each agent should run by using the --count argument.

• PyTorch Lightning CIFAR10 Tutorial
• Wandb Sweep using hydra