from local

This commit is contained in:
dohe0342 2023-02-02 14:01:31 +09:00
parent 51f78f9b8d
commit 9e816d6351
2 changed files with 1 additions and 1 deletions

View File

@ -113,7 +113,7 @@ class Transformer(nn.Module):
# num_layers=num_encoder_layers,
# norm=encoder_norm,
#)
self.encoder = TransformerEncoder(
self.encoder = TransfEncoder(
encoder_layer=encoder_layer,
num_layers=encoder_layer,
norm=encoder_norm,