diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp index 4865a2fdc..9bb86225b 100644 Binary files a/egs/librispeech/ASR/incremental_transf/.model.py.swp and b/egs/librispeech/ASR/incremental_transf/.model.py.swp differ diff --git a/egs/librispeech/ASR/incremental_transf/model.py b/egs/librispeech/ASR/incremental_transf/model.py index 5410c3f76..e12874a6d 100644 --- a/egs/librispeech/ASR/incremental_transf/model.py +++ b/egs/librispeech/ASR/incremental_transf/model.py @@ -234,6 +234,7 @@ class Interformer(nn.Module): super().__init__() assert isinstance(pt_encoder, EncoderInterface), type(pt_encoder) assert isinstance(inter_encoder, EncoderInterface), type(inter_encoder) + self.pt_encoder = pt_encoder self.inter_encoder = inter_encoder