__pycache__/ pretrained_weights/ output/ .venv/ mlruns/ data/ *.pth *.pt *.pkl *.bin