Update pdl.py

ChongQingNoSubway · Apr 28, 2024 · 82b46cb · 82b46cb
1 parent 02a59f8
commit 82b46cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MNIST/pdl.py b/MNIST/pdl.py
@@ -22,7 +22,7 @@ def forward(self,input):
         else:
             #input_2 = self.embedding(input)
             #print(input.shape)
-            importances = torch.mean(input,dim=1,keepdim=True)
+            importances = torch.mean(input,dim=1)
             #importances = torch.sigmoid(importances)
             importances = F.softmax(importances,dim=0)
             #print(importances)