from local

This commit is contained in:
dohe0342 2023-01-09 19:22:03 +09:00
parent 9fdf074007
commit 550d08acd9
2 changed files with 2 additions and 1 deletions

View File

@ -234,7 +234,8 @@ class Interformer(nn.Module):
super().__init__() super().__init__()
assert isinstance(pt_encoder, EncoderInterface), type(pt_encoder) assert isinstance(pt_encoder, EncoderInterface), type(pt_encoder)
assert isinstance(inter_encoder, EncoderInterface), type(inter_encoder) assert isinstance(inter_encoder, EncoderInterface), type(inter_encoder)
self.pt_encoder = pt_encoder
self.inter_encoder = inter_encoder
def forward( def forward(
self, self,