from local

This commit is contained in:
dohe0342 2023-02-02 13:45:34 +09:00
parent 1037c76b31
commit f8af9f0400
2 changed files with 7 additions and 2 deletions

View File

@ -109,9 +109,14 @@ class Transformer(nn.Module):
else: else:
encoder_norm = None encoder_norm = None
self.encoder = nn.TransformerEncoder( #self.encoder = nn.TransformerEncoder(
# encoder_layer=encoder_layer,
# num_layers=num_encoder_layers,
# norm=encoder_norm,
#)
self.encoder = TransformerEncoder(
encoder_layer=encoder_layer, encoder_layer=encoder_layer,
num_layers=num_encoder_layers, num_layers=encoder_layer,
norm=encoder_norm, norm=encoder_norm,
) )