taodeng / CDNN-traffic-saliency

15 stars 8 forks source link

+ Dockerfile and requirements.txt to reproduce experiments #1

Open javiermcebrian opened 4 years ago

javiermcebrian commented 4 years ago

I've developed this Dockerfile and requirements.txt files to be able to execute your code in the environment described in your README.md (python 2, cuda 9.0, cudnn 7, pytorch 0.4.1 and python deps).