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

Disabled Softmax

Browse Source
This commit is contained in:
Christian Risi
2025-10-09 11:36:56 +02:00
parent 1f9c30b531
commit ba592c3480
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Project_Model/Libs/Transformer/Classes/DeToken.py
View File

@@ -14,6 +14,6 @@ class DeToken(torch.nn.Module):
x = self.__linear(x) x = self.__linear(x)
# 2) Go to logits # 2) Go to logits
x = torch.softmax(x, 2) # x = torch.softmax(x, 2)
return x return x