fix dropout bug

Author: yifanfeng97

SuperMoon/models/CNN_HGNN.py

@@ -36,7 +36,7 @@ def forward(self, x):
         for hyconv in self.hyconvs:
             x = hyconv(x, H)
             x = F.leaky_relu(x, inplace=True)
-            x = F.dropout(x, self.dropout)
+            x = F.dropout(x, self.dropout, training=self.training)
         # N x C -> 1 x C x N
         x = x.permute(1, 0).unsqueeze(0)
 

SuperMoon/models/HGNN.py

@@ -19,7 +19,7 @@ def __init__(self, in_ch, n_class, hiddens=[16], dropout=0.5) -> None:
 
     def forward(self, x, H, hyedge_weight=None):
         for hyconv in self.hyconvs:
-            x = F.dropout(x, self.dropout)
+            x = F.dropout(x, self.dropout, training=self.training)
             x = hyconv(x, H)
             x = F.leaky_relu(x, inplace=True)
         x = self.last_hyconv(x, H)