Hey, is there a way on getting torchviz to show the layer names while using nn.Module,
All the examples are I’ve found are using nn.Sequential! for example here 3 where the blue boxes show W0.weight, W0.bias etc.