Is There any way to save nnvm symbol?


#1

I just try the yolo example,

sym, params = from_darknet (....)

Is there any way to save the nnvm symbol just like saving symbol to json file in mxnet

or I can only compile it as a module and export ?


#2
graph = nnvm.graph.create(sym)
with open('my_model.json', 'w') as f:
   f.write(graph.json())

You can easily produce JSON from an NNVM graph and store/load it using traditional APIs.


#3

Thanks for reply, but how to Load Graph (json files) to nnvm symbol ?


#4

nnvm.graph.load_json gives you a nnvm graph, from graph you can fetch symbol.


#5

Thank you very much, I just saw that nnvm.graph got a property called symbol