from local

This commit is contained in:
dohe0342 2023-02-02 14:13:37 +09:00
parent c4b071ab8a
commit 77ac9fb05f
2 changed files with 2 additions and 0 deletions

View File

@ -502,6 +502,8 @@ class TransfEncoder(nn.TransformerEncoder):
if self.norm is not None: if self.norm is not None:
output = self.norm(output) output = self.norm(output)
layer_outputs[-1] = output
return output, layer_outputs return output, layer_outputs