Understanding TensorboardX graph

I’m inspecting my model in the tensorboardX. Following is the forward method:

    def forward(self, x):
        x = self.net(x)
        x = x.view(x.shape[0], -1)  # flat
        x = self.ext_net(x)
        x = F.log_softmax(x, dim=1)

        return x

In the graph, I don’t understand what these values and nodes in blue and green boxes represent. Why is there a separate dataflow from input to ext_net (FCNCustom)? Shouldn’t there be only one coming through net?

enter image description here

1 Like