To set up an environment for this repo, run the following:

conda env create --file environment.yml

To activate this environment, run the following:

conda activate wandb

Once you activate the environment, run main.py, as shown below:

python main.py

Provide your W&B API key when prompted.