Relax OpenFace error bar (#673)

onnx · Dec 21, 2020 · 3b0ccec · 3b0ccec
1 parent 7b613c9
commit 3b0ccec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/applications/nightly_build/test_open_face.py b/applications/nightly_build/test_open_face.py
@@ -268,7 +268,7 @@ def test_OpenFace(self):
         x = np.random.rand(2, 96, 96, 3).astype(np.float32)
         expected = keras_model.predict(x)
         onnx_model = keras2onnx.convert_keras(keras_model, keras_model.name)
-        self.assertTrue(run_onnx_runtime(onnx_model.graph.name, onnx_model, x, expected, self.model_files))
+        self.assertTrue(run_onnx_runtime(onnx_model.graph.name, onnx_model, x, expected, self.model_files, rtol=5e-3, atol=5e-6))
 
 
 if __name__ == "__main__":