models**/
wandb/
data**/
**/__pycache__