3 ms·
Not an expert on ML model, what does the smallest box represent exactly? Function calls?
by TheBigRoomXXL 3y ago
Not an expert on ML model, what does the smallest box represent exactly? Function calls?
- freedmand 3y agoEach 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.html https://pytorch.org/docs/stable/generated/torch.nn.Module.ht...