Open
Description
Not to shamelessly plug and add another package to the pile, but I wrote a package, TorchLens, that can visualize networks in just one line of code (just feed in the network and a sample input and you're good to go). It's meant to work for any PyTorch model whatsoever and has been tested on hundreds of different architectures. Hoping it helps folks out :]
Metadata
Metadata
Assignees
Labels
No labels