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