diff --git a/graphgallery/gallery/nodeclas/dgl/grand.py b/graphgallery/gallery/nodeclas/dgl/grand.py
index ccba7412..988c2edd 100644
--- 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
index b0d66ce0..574c2de5 100644
--- 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__()