From d362a3dba7a19651d268e98a38c3dc10d72c4956 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 18 Dec 2021 11:25:33 +0800
Subject: [PATCH] Reduce the number of decoder layers from 4 to 2.

---
 egs/librispeech/ASR/transducer/decode.py     | 2 +-
 egs/librispeech/ASR/transducer/export.py     | 2 +-
 egs/librispeech/ASR/transducer/pretrained.py | 2 +-
 egs/librispeech/ASR/transducer/train.py      | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/egs/librispeech/ASR/transducer/decode.py b/egs/librispeech/ASR/transducer/decode.py
index eeceb6db2..752712829 100755
--- a/egs/librispeech/ASR/transducer/decode.py
+++ b/egs/librispeech/ASR/transducer/decode.py
@@ -131,7 +131,7 @@ def get_params() -> AttributeDict:
             "vgg_frontend": False,
             # decoder params
             "decoder_embedding_dim": 1024,
-            "num_decoder_layers": 4,
+            "num_decoder_layers": 2,
             "decoder_hidden_dim": 512,
             "env_info": get_env_info(),
         }
diff --git a/egs/librispeech/ASR/transducer/export.py b/egs/librispeech/ASR/transducer/export.py
index 819d2e31d..c74700feb 100755
--- a/egs/librispeech/ASR/transducer/export.py
+++ b/egs/librispeech/ASR/transducer/export.py
@@ -121,7 +121,7 @@ def get_params() -> AttributeDict:
             "vgg_frontend": False,
             # decoder params
             "decoder_embedding_dim": 1024,
-            "num_decoder_layers": 4,
+            "num_decoder_layers": 2,
             "decoder_hidden_dim": 512,
             "env_info": get_env_info(),
         }
diff --git a/egs/librispeech/ASR/transducer/pretrained.py b/egs/librispeech/ASR/transducer/pretrained.py
index 2374de311..f27938de6 100755
--- a/egs/librispeech/ASR/transducer/pretrained.py
+++ b/egs/librispeech/ASR/transducer/pretrained.py
@@ -118,7 +118,7 @@ def get_params() -> AttributeDict:
             "vgg_frontend": False,
             # decoder params
             "decoder_embedding_dim": 1024,
-            "num_decoder_layers": 4,
+            "num_decoder_layers": 2,
             "decoder_hidden_dim": 512,
             "env_info": get_env_info(),
         }
diff --git a/egs/librispeech/ASR/transducer/train.py b/egs/librispeech/ASR/transducer/train.py
index 2ce2fadaa..fe3cc386b 100755
--- a/egs/librispeech/ASR/transducer/train.py
+++ b/egs/librispeech/ASR/transducer/train.py
@@ -198,7 +198,7 @@ def get_params() -> AttributeDict:
             "vgg_frontend": False,
             # decoder params
             "decoder_embedding_dim": 1024,
-            "num_decoder_layers": 4,
+            "num_decoder_layers": 2,
             "decoder_hidden_dim": 512,
             # parameters for Noam
             "warm_step": 80000,  # For the 100h subset, use 8k