1, Clone the repo
2, Install the requirments found in requirements.txt
When installing pytorch, you have to install the Gpu version
pip3 install torch torchvision torchaudio --extra-index-url
You will also need to download the CUDA Toolkit from the NVIDIA website in order to get the CNN models to train on the GPU instead of the CPU
3, Run main.py