from local

This commit is contained in:
dohe0342 2023-01-09 19:47:30 +09:00
parent e8c4d4fe6c
commit 21ef5e476c
3 changed files with 4 additions and 1 deletions

View File

@ -512,7 +512,10 @@ class Tempformer(EncoderInterface):
):
layer_outputs = []
for enum, encoder in enumerate(self.encoder_layers):
layer_outputs.append(encoder(x[enum]))
x_ = x[enum].permute(1, 0, 2)
encoder_output = encoder(x_)
encoder_output = encoder_output.permute(1, 0, 2)
layer_outputs.append(encoder(x_))
return layer_outputs