from local

This commit is contained in:
dohe0342 2023-01-09 19:47:36 +09:00
parent 21ef5e476c
commit 6e789c319f
2 changed files with 1 additions and 1 deletions

View File

@ -515,7 +515,7 @@ class Tempformer(EncoderInterface):
x_ = x[enum].permute(1, 0, 2) x_ = x[enum].permute(1, 0, 2)
encoder_output = encoder(x_) encoder_output = encoder(x_)
encoder_output = encoder_output.permute(1, 0, 2) encoder_output = encoder_output.permute(1, 0, 2)
layer_outputs.append(encoder(x_)) layer_outputs.append(encoder_output)
return layer_outputs return layer_outputs