**/__pycache__/**
/output/two_stage_encoding
*.pt
*.onnx
/output/debug_run
/src/__pycache__
.conda 
.github
.vscode
/src/__pycache__
/src/stage1/__pycache__