wandb-cleanup

Deletes models without tags

To install in your python environment

git clone https://github.com/DuaneNielsen/wandb-cleanup.git
pip install -e .

To use, tag models you want to keep in wandb (by default wandb tags the latest model uploaded) then run

wandb-cleanup <userid> <project_name> 

this will list the model artifacts, and indicate which are to be deleted vs kept

To actually delete execute

wandb-cleanup <userid> <project_name> --delete

example usage

> wandb-cleanup duannielsen oardm_cifar

KEEPING model-2xaf8kp7:v12 with tags ['latest', 'best']
DELETING model-2xaf8kp7:v11
DELETING model-2xaf8kp7:v10
DELETING model-2xaf8kp7:v9
this was a dry run, add --delete to confirm you want to delete/keep the above files

> wandb-cleanup duannielsen oardm_cifar --delete

KEEPING model-2xaf8kp7:v12 with tags ['latest', 'best']
DELETING model-2xaf8kp7:v11
DELETING model-2xaf8kp7:v10
DELETING model-2xaf8kp7:v9