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

MisaOgura / flashtorch

Visualization toolkit for neural networks in PyTorch! Demo -->

https://youtu.be/18Iw4qYqfPo

MIT License

735 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