.venv/ __pycache__/ checkpoints/ dataset/data/