from local

This commit is contained in:
dohe0342 2023-01-09 20:35:24 +09:00
parent 186d756804
commit 751a6fc23a
2 changed files with 1 additions and 0 deletions

View File

@ -503,6 +503,7 @@ class Tempformer(EncoderInterface):
self, x self, x
): ):
layer_outputs = [] layer_outputs = []
pos_emb = x[-1]
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_)