From b82a505dfc003ea9b919dc49d60d780837e927bd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 22 Mar 2022 12:30:48 +0800 Subject: [PATCH] Reduce initial pruned_loss scale from 0.01 to 0.0 --- egs/librispeech/ASR/pruned_transducer_stateless2/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index d28a8a060..b9409127e 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -496,7 +496,7 @@ def compute_loss( warmup_mode=warmup_mode, ) loss = (params.simple_loss_scale * simple_loss + - (pruned_loss * 0.01 if warmup_mode else pruned_loss)) + (pruned_loss * 0.0 if warmup_mode else pruned_loss)) assert loss.requires_grad == is_training