mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 23:54:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2022-05-05 12:35:41 +08:00
parent 03cc9eae9d
commit ce885c6a67
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/librispeech/ASR/pruned_transducer_stateless2/train.py
View File

@ -156,15 +156,16 @@ def get_parser():
"--initial-lr", "--initial-lr",
type=float, type=float,
default=0.003, default=0.003,
help="The initial learning rate. This value should not need to be changed.", help="The initial learning rate. This value should not need to "
"be changed.",
) )
parser.add_argument( parser.add_argument(
"--lr-batches", "--lr-batches",
type=float, type=float,
default=5000, default=5000,
help="""Number of steps that affects how rapidly the learning rate decreases. help="""Number of steps that affects how rapidly the learning rate
We suggest not to change this.""", decreases. We suggest not to change this.""",
) )
parser.add_argument( parser.add_argument(
@ -962,7 +963,6 @@ def display_and_save_batch(
supervisions = batch["supervisions"] supervisions = batch["supervisions"]
features = batch["inputs"] features = batch["inputs"]
cuts = supervisions["cut"]
logging.info(f"features shape: {features.shape}") logging.info(f"features shape: {features.shape}")