aboutsummaryrefslogtreecommitdiff
path: root/tests/DeepSpeechV1InferenceTest.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/DeepSpeechV1InferenceTest.hpp')
-rwxr-xr-xtests/DeepSpeechV1InferenceTest.hpp7
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;