From d789f151c5d5fa84b7595ec5b1d4c99f1e5a2448 Mon Sep 17 00:00:00 2001 From: ftshijt Date: Tue, 23 Nov 2021 09:19:40 -0500 Subject: [PATCH] fix new argument from fairseq --- espnet2/asr/encoder/hubert_encoder.py | 2 ++ 1 file changed, 2 insertions(+) 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()