Update train.py (#1299)

k2-fsa · Oct 11, 2023 · 0d09a44 · 0d09a44
1 parent 9af144c
commit 0d09a44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/aishell/ASR/pruned_transducer_stateless7/train.py b/egs/aishell/ASR/pruned_transducer_stateless7/train.py
@@ -703,7 +703,7 @@ def compute_loss(
             if batch_idx_train >= warm_step
             else 0.1 + 0.9 * (batch_idx_train / warm_step)
         )
-        loss = params.simple_loss_scale * simple_loss + pruned_loss_scale * pruned_loss
+        loss = simple_loss_scale * simple_loss + pruned_loss_scale * pruned_loss
 
     assert loss.requires_grad == is_training