from local

This commit is contained in:
dohe0342 2023-02-25 16:31:19 +09:00
parent 6bcdd98e46
commit 5676e4b756
2 changed files with 1 additions and 0 deletions

View File

@ -883,6 +883,7 @@ class TransformerDecoder(nn.Module):
outputs.append(output)
output = self.combiner(outputs)
print(output.size())
return output