tflite model compilation fails due to missing module "graph_optimizer"
any suggestions to resolve this?
I assume something is wrong in my docker image - if you have is a working image/container, perhaps you can share it on dockerhub?
For what it's worth, the basic example also fails for me with the same error.
traceback and pip dependencies below:
root@b7335b00dd25:/drp-ai_tvm/tutorials# python3 compile_tflite_model.py ../data/model.tflite -o model_tflite
[Check arguments]
Input AI model : ../data/model.tflite
SDK path : /opt/poky/3.1.14
DRP-AI Translator path : /opt/drp-ai_translator_release
Output dir : model_tflite
Default input shape : [1, 3, 224, 224]
-------------------------------------------------
Run TVM frotend compiler
-------------------------------------------------
Run TVM backend compiler with DRP-AI Translator
[03:24:54] /root/drp-tvm-share/src/relay/transforms/to_mixed_precision.cc:429: Warning: Op "layout_transform" not registered FTVMMixedPrecisionConversionType appears 1 times in graph.
def @tvmgen_default_tvmgen_default_mera_drp_91...
One or more operators have not been tuned. Please tune your model for better performance. Use DEBUG logging level to see more details.
Function name: tvmgen_default_tvmgen_default_mera_drp_104
Model size: 5432
GraphSize: 5432
Traceback (most recent call last):
File "compile_tflite_model.py", line 86, in <module>
drp.build(mod, \
File "python/tvm/relay/mera/drp/build.py", line 237, in tvm.relay.mera.drp.build.build
File "python/tvm/relay/mera/drp/build.py", line 238, in tvm.relay.mera.drp.build.build
File "/drp-ai_tvm/tvm/python/tvm/relay/build_module.py", line 357, in build
executor_config, runtime_mod, params = bld_mod.build(
File "/drp-ai_tvm/tvm/python/tvm/relay/build_module.py", line 172, in build
self._build(mod, target, target_host, executor, mod_name)
File "/drp-ai_tvm/tvm/python/tvm/_ffi/_ctypes/packed_func.py", line 237, in __call__
raise get_last_ffi_error()
tvm._ffi.base.TVMError: Traceback (most recent call last):
File "/root/drp-tvm-share/src/relay/backend/contrib/mera/drp/mera_drp_codegen.cc", line 1298
TVMError: drp translator toolchain failed:
Traceback (most recent call last):
File "/opt/drp-ai_translator_release/DRP-AI_translator/api_translator/scripts/run_translator.py", line 1, in <module>
from translator import translator
File "translator.py", line 8, in init translator
ModuleNotFoundError: No module named 'graph_optimizer'
Error: run_translator.py failed
tflite model compilation fails due to missing module "graph_optimizer"
any suggestions to resolve this?
I assume something is wrong in my docker image - if you have is a working image/container, perhaps you can share it on dockerhub?
For what it's worth, the basic example also fails for me with the same error.
traceback and pip dependencies below:
pip environment is: