fixed deprecated Sigmoid

LostOxygen · Jul 3, 2023 · 4bd4aca · 4bd4aca
1 parent d8f2391
commit 4bd4aca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/kernel_eval/models.py b/kernel_eval/models.py
@@ -60,7 +60,7 @@ def forward(self, x_val: Tensor) -> Tensor:
         if self.is_cifar:
             out = F.softmax(out, dim=-1)
         else:
-            out = F.sigmoid(out)
+            out = torch.sigmoid(out)
 
         return out
 
@@ -251,7 +251,7 @@ def forward(self, x_val: Tensor) -> Tensor:
         if self.is_cifar:
             out = F.softmax(out, dim=-1)
         else:
-            out = F.sigmoid(out)
+            out = torch.sigmoid(out)
 
         return out
 
@@ -294,6 +294,6 @@ def forward(self, out) -> torch.Tensor:
         if self.is_cifar:
             out = F.softmax(out, dim=-1)
         else:
-            out = F.sigmoid(out)
+            out = torch.sigmoid(out)
 
         return out