from local
This commit is contained in:
parent
51f78f9b8d
commit
9e816d6351
Binary file not shown.
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user