Show HN: Library to visualize and inspect PyTorch models (opens in new tab)

(github.com)

4 pointsfreedmand2y ago2 comments

2 comments

Not an expert on ML model, what does the smallest box represent exactly? Function calls?

Each box is a torch module[1], which can technically encapsulate any amount of computation or contain sub-modules. The smallest boxes or leaf nodes are those that contain no sub-modules. They can still run an arbitrary number of functions depending on the node.

[1] https://pytorch.org/docs/stable/generated/torch.nn.Module.ht...

j / k navigate · click thread line to collapse

2 comments

TheBigRoomXXL2y ago

Not an expert on ML model, what does the smallest box represent exactly? Function calls?

freedmandOP2y ago

[1] https://pytorch.org/docs/stable/generated/torch.nn.Module.ht...

j / k navigate · click thread line to collapse