fix new argument from fairseq

shinichiye · Nov 23, 2021 · d789f15 · d789f15
1 parent 0d8cd47
commit d789f15
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/espnet2/asr/encoder/hubert_encoder.py b/espnet2/asr/encoder/hubert_encoder.py
@@ -267,6 +267,7 @@ def __init__(
         activation_dropout_rate: float = 0.0,
         hubert_dict: str = "./dict.txt",
         label_rate: int = 100,
+        checkpoint_activations: bool = false,
         sample_rate: int = 16000,
         use_amp: bool = False,
         **kwargs,
@@ -296,6 +297,7 @@ def __init__(
             "dropout": dropout_rate,
             "attention_dropout": attention_dropout_rate,
             "label_rate": label_rate,
+            "checkpoint_activations": checkpoint_activations,
         }
         cfg_overides = {**cfg_overides, **kwargs}
         self.cfg = HubertConfig()