diff --git a/espnet2/asr/encoder/hubert_encoder.py b/espnet2/asr/encoder/hubert_encoder.py index 1664288f697..4f369f1760b 100644 --- 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()