15 Jan 2025

Adding wandb to a CLI yolo run

Assuming you’re doing a YOLO run w/

yolo detect train model=yolo11s.pt data=/data/data/data.yaml project=/data/project/ epochs=500 imgsz=640 device=0,1 name=yolo11s-aug-500epochs-full

name there becomes training run name in wandb + directory name in /data/projct
- (project on wandb will be `-data-project-)
To enable wandb:

pip install wandb
yolo settings wandb=True
wandb login

Or if you’re inside an ultralytics:latest Docker container,

apt install -y bash screen
bash
pip install wandb
yolo settings wandb=True
wandb login 
screen 
yolo detect train model=yolo11s.pt data=/data/data/data.yaml project=/data/project/ epochs=500 imgsz=640 device=0,1 name=yolo11s-aug-500epochs-full

Also useful:

# get a model file
wandb artifact get /proje:ject/run_alxxxpy7_model:v0 --root target_director/

Nel mezzo del deserto posso dire tutto quello che voglio.

comments powered by Disqus