issues
search
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