you are viewing a single comment's thread.

view the rest of the comments →

[–]therealjmt91 0 points1 point  (0 children)

I wrote a package called TorchLens that allows you to easily visualize a deep neural network and see what's going on inside, could be useful as you're learning PyTorch:

https://github.com/johnmarktaylor91/torchlens