Update dropout.py

ChongQingNoSubway · Apr 28, 2024 · 221528e · 221528e
1 parent 82b46cb
commit 221528e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/WSI/models/dropout.py b/WSI/models/dropout.py
@@ -32,7 +32,7 @@ def generate_mask(self,importance,input):
         #print(interpolation)
         mask = torch.zeros_like(importance)
         mask = mask.to(input.device)
-        _, indx = torch.sort(importance,dim=0,descending=True)
+        _, indx = torch.sort(importance,dim=0)
         #print(indx)
         idx = indx.view(-1)
         mask.index_add_(0,idx,interpolation)