From 8ba98cb518d494a2f7a63a748cf7f8a82759c3bc Mon Sep 17 00:00:00 2001
From: Yibing Liu <liuyibing01@baidu.com>
Date: Tue, 27 Jun 2017 18:35:49 +0800
Subject: [PATCH] fix decoders' unittest

---
 infer.py               | 2 +-
 tests/test_decoders.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/infer.py b/infer.py
index 5f0f268a..686f2822 100644
--- a/infer.py
+++ b/infer.py
@@ -63,7 +63,7 @@ parser.add_argument(
     help="Manifest path for decoding. (default: %(default)s)")
 parser.add_argument(
     "--model_filepath",
-    default='checkpoints/params.tar.gz.41',
+    default='checkpoints/params.latest.tar.gz',
     type=str,
     help="Model filepath. (default: %(default)s)")
 parser.add_argument(
diff --git a/tests/test_decoders.py b/tests/test_decoders.py
index 4435355c..a5e19b08 100644
--- a/tests/test_decoders.py
+++ b/tests/test_decoders.py
@@ -81,7 +81,8 @@ class TestDecoders(unittest.TestCase):
             probs_split=[self.probs_seq1, self.probs_seq2],
             beam_size=self.beam_size,
             vocabulary=self.vocab_list,
-            blank_id=len(self.vocab_list))
+            blank_id=len(self.vocab_list),
+            num_processes=24)
         self.assertEqual(beam_results[0][0][1], self.beam_search_result[0])
         self.assertEqual(beam_results[1][0][1], self.beam_search_result[1])
 
-- 
GitLab