-
Notifications
You must be signed in to change notification settings - Fork 19
Closed
Description
I want to convert this model to ncnn but failed. I found that the self.model_config['config'] in this model is a list, which caused error.
Traceback (most recent call last):
File "/home/tommy/miniconda3/envs/pytorch/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/tommy/miniconda3/envs/pytorch/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/home/tommy/miniconda3/envs/pytorch/lib/python3.8/site-packages/keras2ncnn/__main__.py", line 12, in <module>
sys.exit(_main())
File "/home/tommy/miniconda3/envs/pytorch/lib/python3.8/site-packages/keras2ncnn/keras2ncnn.py", line 55, in main
H5dfParser(args.input_file).parse_graph(keras_graph)
File "/home/tommy/miniconda3/envs/pytorch/lib/python3.8/site-packages/keras2ncnn/h5df_parser.py", line 111, in parse_graph
for layers in self.model_config['config']['layers']:
TypeError: list indices must be integers or slices, not str
Metadata
Metadata
Assignees
Labels
No labels