diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index 9f73c8fbc..83558a72b 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -160,7 +160,7 @@ def get_parser(): parser.add_argument( "--num-lr-decays", - type=float, + type=int, default=4, help="The total number of times we decay (halve) the learning rate" )