def create_model(self):
mp = ModelProto()
mp.ir_version = ONNX_IR_VERSION
op = mp.opset_import.add()
op.domain = "" # empty string indicates ONNX domain
op.version = ONNX_OPSET_VERSION
# nn_opset = mp.opset_import.add()
# nn_opset.domain = NNABLA_DOMAIN
# nn_opset.version = NNABLA_OPSET_VERSION
mp.producer_name = PRODUCER_NAME
mp.producer_version = PRODUCER_VERSION
mp.domain = NNABLA_DOMAIN
self._model_proto = mp
def nnp_model_to_onnx_protobuf(nnp, batch_size):
mp = ModelProto()
mp.ir_version = ONNX_IR_VERSION
op = mp.opset_import.add()
op.domain = "" # empty string indicates ONNX domain
op.version = ONNX_OPSET_VERSION
# nn_opset = mp.opset_import.add()
# nn_opset.domain = NNABLA_DOMAIN
# nn_opset.version = NNABLA_OPSET_VERSION
mp.producer_name = PRODUCER_NAME
mp.producer_version = PRODUCER_VERSION
mp.domain = NNABLA_DOMAIN
nnp_model_to_onnx_graph(mp.graph, nnp, batch_size)
return mp
