model = load_model(‘mnist.model’)
print(model.summary())
#dtype = ‘uint8’
dtype = ‘float32’
target=‘llvm’
target_host = ‘llvm’
layout = None
ctx = tvm.cpu(0)
shapes = (1,1,28,28)
out_path = ‘/home/arjun1/ARJUN/my_tvm/out’
sym, params = nnvm.frontend.from_keras(model)
ERROR
Traceback (most recent call last):
File “create_so_mnist_keras.py”, line 37, in
sym, params = nnvm.frontend.from_keras(model)
File “/home/arjun1/ARJUN/my_tvm/tvm/nnvm/python/nnvm/frontend/keras.py”, line 690, in from_keras
sym = symtab.get_var(layer.name, must_contain=True)
File “/home/arjun1/ARJUN/my_tvm/tvm/nnvm/python/nnvm/frontend/common.py”, line 158, in get_var
assert name in self.vars
AssertionError