.venv/ .mypy_cache/ __pycache__/ wandb/ images/ checkpoints/ *.pth *.onnx *.png *.jpg