reneepc's picture
Change model to run on cuda
987d5d0