.venv/
.mypy_cache/
__pycache__/

wandb/
images/

*.pth
*.png
*.jpg