tmp/
checkpoints/
trained_model/
__pycache__/