From 0a5b639ec1cd3bd292b8ae84210f171c0355dcda Mon Sep 17 00:00:00 2001
From: glynpu <guonpu@qq.com>
Date: Fri, 17 Mar 2023 12:17:08 +0800
Subject: [PATCH] update comments

---
 egs/himia/wuw/ctc_tdnn/asr_datamodule.py | 20 ++++++--------------
 1 file changed, 6 insertions(+), 14 deletions(-)

diff --git a/egs/himia/wuw/ctc_tdnn/asr_datamodule.py b/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
index 785e06a7c..565039062 100644
--- a/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
+++ b/egs/himia/wuw/ctc_tdnn/asr_datamodule.py
@@ -16,7 +16,6 @@
 
 
 import argparse
-import inspect
 import logging
 from functools import lru_cache
 from pathlib import Path
@@ -190,7 +189,7 @@ class HiMiaWuwDataModule:
             "--input-strategy",
             type=str,
             default="PrecomputedFeatures",
-            help="PrecomputedFeatures",
+            help="AudioSamples or PrecomputedFeatures",
         )
         group.add_argument(
             "--train-channel",
@@ -198,6 +197,8 @@ class HiMiaWuwDataModule:
             default="_7_01",
             help="""channel of HI_MIA train dataset.
             All channels are used if it is set "all".
+            Please refer state 6 in prepare.sh for its meaning and other
+            potential values. Currently, Only "_7_01" is verified.
             """,
         )
         group.add_argument(
@@ -206,6 +207,8 @@ class HiMiaWuwDataModule:
             default="_7_01",
             help="""channel of HI_MIA dev dataset.
             All channels are used if it is set "all".
+            Please refer state 6 in prepare.sh for its meaning and other
+            potential values. Currently, Only "_7_01" is verified.
             """,
         )
 
@@ -248,22 +251,11 @@ class HiMiaWuwDataModule:
 
         input_transforms = []
         if self.args.enable_spec_aug:
-            logging.info("Enable SpecAugment")
             logging.info(f"Time warp factor: {self.args.spec_aug_time_warp_factor}")
-            # Set the value of num_frame_masks according to Lhotse's version.
-            # In different Lhotse's versions, the default of num_frame_masks is
-            # different.
-            num_frame_masks = 10
-            num_frame_masks_parameter = inspect.signature(
-                SpecAugment.__init__
-            ).parameters["num_frame_masks"]
-            if num_frame_masks_parameter.default == 1:
-                num_frame_masks = 2
-            logging.info(f"Num frame mask: {num_frame_masks}")
             input_transforms.append(
                 SpecAugment(
                     time_warp_factor=self.args.spec_aug_time_warp_factor,
-                    num_frame_masks=num_frame_masks,
+                    num_frame_masks=10,
                     features_mask_size=27,
                     num_feature_masks=2,
                     frames_mask_size=100,