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):
|
for enum, encoder in enumerate(self.encoder_layers):
|
||||||
x_ = x[enum].permute(1, 0, 2)
|
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)
|
encoder_output = encoder_output.permute(1, 0, 2)
|
||||||
layer_outputs.append(encoder_output)
|
layer_outputs.append(encoder_output)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user