From d2db7063840eb38557e44766a2bc9feb80d9a6f9 Mon Sep 17 00:00:00 2001
From: huangyuxin <huangyuxin@baidu.com>
Date: Thu, 22 Jul 2021 02:56:52 +0000
Subject: [PATCH] added autolog

---
 deepspeech/exps/deepspeech2/model.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/deepspeech/exps/deepspeech2/model.py b/deepspeech/exps/deepspeech2/model.py
index 3810694a..54ad449c 100644
--- a/deepspeech/exps/deepspeech2/model.py
+++ b/deepspeech/exps/deepspeech2/model.py
@@ -35,12 +35,12 @@ from deepspeech.utils import error_rate
 from deepspeech.utils import layer_tools
 from deepspeech.utils import mp_tools
 from deepspeech.utils.log import Log
-logger = Log(__name__).getlog()
 
 import auto_log
 import os
 from paddle import inference
 
+logger = Log(__name__).getlog()
 
 
 class DeepSpeech2Trainer(Trainer):
@@ -228,13 +228,12 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
 
     def __init__(self, config, args):
         super().__init__(config, args)
-        # added by hyx
         pid = os.getpid()
+        gpu_id = int(os.environ['CUDA_VISIBLE_DEVICES'].split(',')[0])
         infer_config = inference.Config()
-        infer_config.enable_use_gpu(10000, 2)
-        logger = None
+        infer_config.enable_use_gpu(100, gpu_id)
         autolog = auto_log.AutoLogger(
-             model_name="tiny_s0",
+             model_name="deepspeech2",
              model_precision="fp32",
              batch_size=config.decoding.batch_size,
              data_shape="dynamic",
@@ -242,13 +241,14 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
              inference_config=infer_config,
              pids=pid,
              process_name=None,
-             gpu_ids=2,
+             gpu_ids=gpu_id,
              time_keys=[
                      'preprocess_time', 'inference_time', 'postprocess_time'
              ],
              warmup=0)
         self.autolog = autolog
         logger = autolog.logger
+        logger.info("gpu_id:{}".format(gpu_id))
 
     def ordid2token(self, texts, texts_len):
         """ ord() id to chr() chr """
@@ -291,6 +291,7 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
         self.autolog.times.stamp()
         self.autolog.times.stamp()
         self.autolog.times.end()
+
         for utt, target, result in zip(utts, target_transcripts,
                                        result_transcripts):
             errors, len_ref = errors_func(target, result)