Skip to content
Snippets Groups Projects
Commit 1bdc56fb authored by shingjan's avatar shingjan
Browse files

change a bit

parent 96c794f8
No related branches found
No related tags found
No related merge requests found
import sys import sys
from onnx import numpy_helper from onnx import numpy_helper
from graph_ir import Node
from tensor import InputTensor, WeightTensor from tensor import InputTensor, WeightTensor
support_onnx_ops = {"DepthwiseConv" : [2], support_onnx_ops = {"DepthwiseConv" : [2],
......
...@@ -41,7 +41,7 @@ def compile(model): ...@@ -41,7 +41,7 @@ def compile(model):
gCodegen.compile() gCodegen.compile()
def main(): def main():
model = onnx.load('../models/keras/resnet.onnx') model = onnx.load('../models/keras/lenet.onnx')
# model = onnx.load('../models/keras/vgg16_cifar10.onnx') # model = onnx.load('../models/keras/vgg16_cifar10.onnx')
compile(model) compile(model)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment