from local

This commit is contained in:
dohe0342 2023-01-09 19:20:09 +09:00
parent ae12df45d6
commit 8bddac0e71
2 changed files with 0 additions and 6 deletions

View File

@ -235,14 +235,8 @@ class Interformer(nn.Module):
"""
super().__init__()
assert isinstance(encoder, EncoderInterface), type(encoder)
assert hasattr(decoder, "blank_id")
self.encoder = encoder
self.decoder = decoder
self.joiner = joiner
self.simple_am_proj = ScaledLinear(encoder_dim, vocab_size, initial_speed=0.5)
self.simple_lm_proj = ScaledLinear(decoder_dim, vocab_size)
def forward(
self,