Make all code properly Device aware

hmc-cs-mdrissi / neural_nets_research

A collection of projects for neural nets research in Mehdi's research group.

MIT License

7 stars 1 forks source link

Make all code properly Device aware #12

Open hmc-cs-mdrissi opened 6 years ago

hmc-cs-mdrissi commented 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.