boom boom

2020-05-25 21:23:39 -04:00
parent 0c3d22ea86
commit 81e939fb54
1 changed files with 1 additions and 1 deletions
@@ -84,7 +84,7 @@ class SummarizationTrainer(BaseTransformer):
        assert self.target_lens["train"] <= self.target_lens["test"], f"target_lens: {self.target_lens}"
        self.n_obs = {k: v if v >= 0 else None for k, v in base_nobs.items()}
        self.freeze_stuff()
-        freeze_part(self.model.model.encoder)
+

    def freeze_stuff(self):