Hello,
I wanted to use hiddenlayer, but I am not sure about the second parameter (torch.zeros([1, 1, 512, 512]).to(device)), how exactly it should look? I think that the last 3 number are channels and size of image, but what exactly is the first number? So far I have implemented it like this:
summary(net, (1, 512, 512))
# Build HiddenLayer graph
hl_graph = hl.build_graph(net, torch.zeros([1, 1, 512, 512]).to(device))
# Use a different color theme
hl_graph.theme = hl.graph.THEMES["blue"].copy() # Two options: basic and blue
hl_graph.save(path=os.path.join(dirname, outputDir) , format="png")
But I'm getting this error
Unsupported: ONNX export of Pad in opset 9. The sizes of the padding must be constant. Please try opset version 11.
Hello, I wanted to use hiddenlayer, but I am not sure about the second parameter (
torch.zeros([1, 1, 512, 512]).to(device)
), how exactly it should look? I think that the last 3 number are channels and size of image, but what exactly is the first number? So far I have implemented it like this:But I'm getting this error
The output from summary seems to work ok: