XiaoMi / mace-models

Mobile AI Compute Engine Model Zoo
Apache License 2.0
371 stars 73 forks source link

how to add multi-inputs? or how to test the benchmark on multi inputs? #28

Open peiwenYe opened 4 years ago

peiwenYe commented 4 years ago

hello, I wonder that how to load multi inputs, I just add a comma and another input follow the original "validation_inputs_data", but a mistake occured ><, so can you answer me about that how to add two or three even more inputs to the validation input? Thank you very much!

lee-bin commented 4 years ago

Does your model have multiple input nodes? Or you want to validate the model with different input data? You can paste the yaml file if it helps to clarify.

peiwenYe commented 4 years ago

Thank you for your reply very much. Here is my yaml, I just want to test multi input with the same input format. If I only put one input like "validation_inputs_data: - /home/ypw/speech/kaldi-onnx-master/mace_input.npy", It works. library_name: aishell2 target_abis: [aarch64-linux-gnu] model_graph_format: file model_data_format: file models: aishell2: platform: onnx model_file_path: /home/ypw/speech/kaldi-onnx-master/text.onnx model_sha256_checksum: 8d60e44c128523a82c82f9d592912fa9fe8923ad81443b66c9932f92d0d1dc3f subgraphs:

lee-bin commented 4 years ago

MACE only supports validation for one group of inputs (for models with multiple input nodes), you can't specify multiple groups of input data. You can validate it multiple times with different groups of input data or write your own codes to do it.

peiwenYe commented 4 years ago

I got it. Thank you very much!