CSE534_Final_Project

Test on Python=3.8, Pytorch=1.12.1, torchvision=0.13.1:

[Optional] .Pytorch can be installed by:
GPU version:

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

CPU only:

 conda install pytorch torchvision torchaudio cpuonly -c pytorch 

Install required package .

pip install -r requirements.txt

Before Training:

1. Download dataset from the google drive
   
2. Create a data folder and unzip the datafiles under the data folder
   
3. Check config file in 'conf' folder.