diff --git a/fairseq/models/hubert/hubert.py b/fairseq/models/hubert/hubert.py index 232a5e402a..412d936543 100644 --- a/fairseq/models/hubert/hubert.py +++ b/fairseq/models/hubert/hubert.py @@ -216,6 +216,9 @@ class HubertConfig(FairseqDataclass): metadata={"help": "skip computing losses over unmasked frames"}, ) + checkpoint_activations: bool = field( + default=False, metadata={"help": "recompute activations and save memory for extra compute"} + ) @register_model("hubert", dataclass=HubertConfig) class HubertModel(BaseFairseqModel):