.vscode/ __pycache__/ models/ Datasets/ test_images/ train_logs/ lightning_logs/ .env