Convert into ONNX
97mik opened this issue · comments
Mihail Verenich commented
How can I export ONNX model from latest_net_G.pth?
Richard Zhang commented
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
Ziniu Wu commented
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.
Jun-Yan Zhu commented
See test.py
and this line is loading the model:
model = create_model(opt)