MisaOgura / flashtorch

Visualization toolkit for neural networks in PyTorch! Demo -->
https://youtu.be/18Iw4qYqfPo
MIT License
734 stars 87 forks source link

use the right device -- avoid error about 2 tensors being on different devices #40

Closed minhlab closed 2 years ago