fix reduction conformer_ctc3/train.py (facebookresearch#908)

yfyeung · Feb 14, 2023 · 4e832fa · 4e832fa
1 parent c5e687d
commit 4e832fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/librispeech/ASR/conformer_ctc3/train.py b/egs/librispeech/ASR/conformer_ctc3/train.py
@@ -382,7 +382,7 @@ def get_params() -> AttributeDict:
             "num_encoder_layers": 12,
             # parameters for loss
             "beam_size": 10,
-            "reduction": "sum",
+            "reduction": "none",
             "use_double_scores": True,
             # parameters for Noam
             "model_warm_step": 3000,  # arg given to model, not for lrate