HoneyTian commited on
Commit
74270c2
·
1 Parent(s): b8a9d61
examples/sound_classification_by_lstm/step_9_evaluation_onnx_model.py CHANGED
@@ -73,9 +73,9 @@ def main():
73
  evaluation_file = tgt_path / "evaluation.xlsx"
74
  else:
75
  config_file = model_dir / "config.yaml"
76
- onnx_model_file = args.onnx_model_file
77
- vocab_path = args.vocabulary_dir
78
- evaluation_file = args.dataset
79
 
80
  config = WaveClassifierConfig.from_pretrained(config_file.as_posix())
81
  ort_session = ort.InferenceSession(onnx_model_file.as_posix())
 
73
  evaluation_file = tgt_path / "evaluation.xlsx"
74
  else:
75
  config_file = model_dir / "config.yaml"
76
+ onnx_model_file = Path(args.onnx_model_file)
77
+ vocab_path = Path(args.vocabulary_dir)
78
+ evaluation_file = Path(args.dataset)
79
 
80
  config = WaveClassifierConfig.from_pretrained(config_file.as_posix())
81
  ort_session = ort.InferenceSession(onnx_model_file.as_posix())