from local
This commit is contained in:
parent
1707909fb6
commit
999912ecc7
Binary file not shown.
@ -507,7 +507,7 @@ class Tempformer(EncoderInterface):
|
||||
|
||||
for enum, encoder in enumerate(self.encoder_layers):
|
||||
x_ = x[enum].permute(1, 0, 2)
|
||||
encoder_output = encoder(x_)
|
||||
encoder_output = encoder(x_, pos_emb)
|
||||
encoder_output = encoder_output.permute(1, 0, 2)
|
||||
layer_outputs.append(encoder_output)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user