Run data preparation create the dataset run py src/data_prep.py to convert all colored input data to grayscale Modules install torch for cuda https://pytorch.org/get-started/locally/ pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118