nullgeppetto:

import torch.onnx
dummy_input = Variable(torch.randn(4, 3, 32, 32))
torch.onnx.export(net, dummy_input, "model.onnx")



Thank u so much…it helped me to visualize my ABCDs of resnet to understand it better.