Also, I noticed that the script at src/lobster/cmdline/_train.py explicitly logs in to genentech.wandb.io in the wandb.login().
Would you be open to parameterizing the host in the src/lobster/hydra_config/logger/wandb.yaml file instead? It would make the repo marginally more usable to newcomers, although I understand if you prefer end-users implement their own train script.