Skip to content
Snippets Groups Projects
config.py 214 B
model_name = "lenet"
compile_type = 1 # 0 for HPVM Tensor Runtime, 1 for HPVM C Interface
input_size = [1,2,3,4]
onnx_file_dir = "../models/keras/lenet.onnx"
opset_version_default = 10
src_emit_dir = "./test_src"