anjalidhabaria / ECE6254_Project_Enet

ECE6254_Project_Enet

Statistical machine learning ECE 6254 project - Scene understanding using ENet

Pytorch 1.2

Cuda 10.0

Installed through Anaconda by following command

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

Also added a notebook. If someone wishes to work on Jupyter