diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp index 5759f180d..4865a2fdc 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 d986156a5..5410c3f76 100644 --- a/egs/librispeech/ASR/incremental_transf/model.py +++ b/egs/librispeech/ASR/incremental_transf/model.py @@ -234,7 +234,8 @@ 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 def forward( self,