from local

This commit is contained in:
dohe0342 2023-02-02 14:07:59 +09:00
parent 89271fb9ea
commit 8a36b40a12
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: if self.norm is not None:
output = self.norm(output) output = self.norm(output)
return output return output, outputs
class TransformerEncoderLayer(nn.Module): class TransformerEncoderLayer(nn.Module):