fix hard coded cpu type

MathGaron / pytorch_toolbox

Boiler plate code for pytorch. Train/Validation loops, visualization etc. For research.

MIT License

10 stars 3 forks source link

Closed MathGaron closed 7 years ago