From a681a8c9313b49860295fda370aae4b3b6b59855 Mon Sep 17 00:00:00 2001
From: jingyong hou <houjingyong@gmail.com>
Date: Sun, 21 Nov 2021 23:20:16 +0800
Subject: [PATCH] [fix bug] resolve bugs in score.py

---
 examples/hi_xiaowen/s0/path.sh |  2 +-
 kws/bin/score.py               | 20 ++++++++++++++++++--
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/examples/hi_xiaowen/s0/path.sh b/examples/hi_xiaowen/s0/path.sh
index cf09584..b90a515 100755
--- a/examples/hi_xiaowen/s0/path.sh
+++ b/examples/hi_xiaowen/s0/path.sh
@@ -2,4 +2,4 @@ export PATH=$PWD:$PATH
 
 # NOTE(kan-bayashi): Use UTF-8 in Python to avoid UnicodeDecodeError when LC_ALL=C
 export PYTHONIOENCODING=UTF-8
-export PYTHONPATH=../../:$PYTHONPATH
+export PYTHONPATH=../../../:$PYTHONPATH
diff --git a/kws/bin/score.py b/kws/bin/score.py
index 0e8c18f..4c80816 100644
--- a/kws/bin/score.py
+++ b/kws/bin/score.py
@@ -43,6 +43,18 @@ def get_args():
                         default=16,
                         type=int,
                         help='batch size for inference')
+    parser.add_argument('--num_workers',
+                        default=0,
+                        type=int,
+                        help='num of subprocess workers for reading')
+    parser.add_argument('--pin_memory',
+                        action='store_true',
+                        default=False,
+                        help='Use pinned memory buffers used for reading')
+    parser.add_argument('--prefetch',
+                        default=100,
+                        type=int,
+                        help='prefetch number')
     parser.add_argument('--score_file',
                         required=True,
                         help='output score file')
@@ -65,11 +77,15 @@ def main():
     test_conf['speed_perturb'] = False
     test_conf['spec_aug'] = False
     test_conf['shuffle'] = False
-    test_conf['fbank_conf']['dither'] = 0.0
+    test_conf['feature_extraction_conf']['dither'] = 0.0
     test_conf['batch_conf']['batch_size'] = args.batch_size
 
     test_dataset = Dataset(args.test_data, test_conf)
-    test_data_loader = DataLoader(test_dataset, batch_size=None, num_workers=0)
+    test_data_loader = DataLoader(test_dataset,
+                                  batch_size=None,
+                                  pin_memory=args.pin_memory,
+                                  num_workers=args.num_workers,
+                                  prefetch_factor=args.prefetch)
 
     # Init asr model from configs
     model = init_model(configs['model'])