PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fixed bugs and added visibility

Browse Source
This commit is contained in:
Christian Risi
2025-10-11 21:49:29 +02:00
parent 160b7dbfc0
commit bcc2fe7368
2 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Project_Model/Libs/Transformer/Models/NanoSocraDecoder.py
View File

@@ -24,7 +24,7 @@ class NanoSocraDecoder(torch.nn.Module):
decoder_tensor = self.__decoder_embedder(decoder_embedder_input)
decoder_output, _, _, _, _, _ = self.__decoder(
(decoder_tensor, decoder_tensor, decoder_tensor, tgt_padding, tgt_padding, False)
(decoder_tensor, decoder_tensor, decoder_tensor, tgt_padding, tgt_padding, True)
)
logits: torch.Tensor = self.__detokener(decoder_output)