Convert into ONNX

[97mik]

How can I export ONNX model from latest_net_G.pth?

[richzhang]

Not sure how to do it, but a more straightforward model file just for execution is in the GUI repository, more specifically here: https://github.com/junyanz/interactive-deep-colorization/blob/master/models/pytorch/model.py

[wuziniu]

I have the same question. How can I load your pretrained model from latest_net_G.pth as nn.module model? The script you refered to is only defining the network structure. Thank you so much.

[junyanz]

See test.py and this line is loading the model:

model = create_model(opt)