from local

This commit is contained in:
dohe0342 2023-02-02 14:10:41 +09:00
parent 369b314d85
commit c4b071ab8a
2 changed files with 1 additions and 1 deletions

View File

@ -502,7 +502,7 @@ class TransfEncoder(nn.TransformerEncoder):
if self.norm is not None:
output = self.norm(output)
return output, outputs
return output, layer_outputs
class TransformerEncoderLayer(nn.Module):