Hi @zsp1197, have you managed to plot the graph in VidualDL? I’m trying to do that for a very simple network, following this 213 example.
What I understand is that I need to export my model in a onnx file, as follows:
import torch.onnx
dummy_input = Variable(torch.randn(4, 3, 32, 32))
torch.onnx.export(net, dummy_input, "model.onnx")

Then, VisualDL’s documentation suggests this 392. However, in such a case I don’t what logdir to use.