Network Architecture of forward method

#1

Hi I am trying to figure out how data flow in forward method

Is there is any way I can visualize this below part.

    def forward(self, x):
        x = self.conv1(x)
        x = self.bn1(x)
        x = self.relu(x)
        x = self.maxpool(x)

        x = self.layer1(x)
        x = self.layer2(x)
        x = self.layer3(x)
        x = self.layer4(x)

        x = self.avgpool(x)
        x = x.view(x.size(0), -1)
        x = self.fc(x)

        return x

Thanks

#2

The data will flow through each module from sequentially.
You could visualize this graph via e.g. torchviz.