__pycache__
checkpoints/
wandb/