As the title suggests, I am wondering if anyone is aware of a library that will create a schematic of your pytorch model?
This is a snip from google of Alexnet, or something similar:
As the title suggests, I am wondering if anyone is aware of a library that will create a schematic of your pytorch model?
This is a snip from google of Alexnet, or something similar: