from local

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

View File

@ -234,6 +234,7 @@ 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.pt_encoder = pt_encoder
self.inter_encoder = inter_encoder self.inter_encoder = inter_encoder