diff options
Diffstat (limited to 'tests/DeepSpeechV1InferenceTest.hpp')
-rwxr-xr-x | tests/DeepSpeechV1InferenceTest.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/DeepSpeechV1InferenceTest.hpp b/tests/DeepSpeechV1InferenceTest.hpp index 24e7dac567..633176219c 100755 --- a/tests/DeepSpeechV1InferenceTest.hpp +++ b/tests/DeepSpeechV1InferenceTest.hpp @@ -124,7 +124,7 @@ public: Model::AddCommandLineOptions(options, m_ModelCommandLineOptions); } - virtual bool ProcessCommandLineOptions() override + virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) override { if (!ValidateDirectory(m_InputSeqDir)) { @@ -156,7 +156,7 @@ public: return false; } - m_Model = m_ConstructModel(m_ModelCommandLineOptions); + m_Model = m_ConstructModel(commonOptions, m_ModelCommandLineOptions); if (!m_Model) { return false; @@ -185,7 +185,8 @@ public: private: typename Model::CommandLineOptions m_ModelCommandLineOptions; - std::function<std::unique_ptr<Model>(typename Model::CommandLineOptions)> m_ConstructModel; + std::function<std::unique_ptr<Model>(const InferenceTestOptions&, + typename Model::CommandLineOptions)> m_ConstructModel; std::unique_ptr<Model> m_Model; std::string m_InputSeqDir; |