Hi, I added a graph and …
The graph shows operations of torch.Tensor.view()
def forward(self, x):
x = self.conv1(x)
x = self.conv2(x)
x = self.pull1(x)
x = self.conv3(x)
x = self.pull2(x)
x = x.view(x.size(0), -1)
x = self.linear(x)
It definitely annoys me… are there any ways to make a clean graph?
Thanks!