Merge branch 'master' of https://github.com/EdisonLeeeee/GraphGallery …

…into master
EdisonLeeeee · Sep 21, 2021 · f391c19 · f391c19
2 parents 194a8ba + a7bb0d8
commit f391c19
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/graphgallery/gallery/nodeclas/dgl/grand.py b/graphgallery/gallery/nodeclas/dgl/grand.py
@@ -29,8 +29,8 @@ def data_step(self,
     def model_step(self,
                    hids=[16],
                    acts=['relu'],
-                   S=4,
-                   K=8, temp=0.5, lam=1.,
+                   S=1,
+                   K=4, temp=0.5, lam=1.,
                    dropout=0.5,
                    weight_decay=5e-4,
                    lr=0.01,

diff --git a/graphgallery/nn/models/dgl/grand.py b/graphgallery/nn/models/dgl/grand.py
@@ -13,7 +13,7 @@
 def drop_node(feats, drop_rate, training):
 
     n = feats.shape[0]
-    drop_rates = torch.FloatTensor(np.ones(n) * drop_rate)
+    drop_rates = torch.ones(n) * drop_rate
 
     if training:
 
@@ -66,12 +66,13 @@ def __init__(self,
                  hids=[16],
                  acts=['relu'],
                  dropout=0.5,
-                 S=4,
-                 K=8,
-                 temp=0.5, lam=1.,
+                 S=1,
+                 K=4,
+                 temp=0.5, 
+                 lam=1.,
                  weight_decay=5e-4,
                  lr=0.01,
-                 bias=True,
+                 bias=False,
                  bn=False):
 
         super().__init__()