search

MisaOgura / flashtorch

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

Backprop.visualize passes use_gpu flag #25

Closed MisaOgura closed 4 years ago

MisaOgura commented 4 years ago

Closes #18