Open hmc-cs-mdrissi opened 6 years ago
Currently our code is fairly messy on how properly supporting both cpu/cuda. The version 4 upgrade of pytorch has added some features to make writing device aware code easier.
Currently our code is fairly messy on how properly supporting both cpu/cuda. The version 4 upgrade of pytorch has added some features to make writing device aware code easier.