modify default lr and fanout.

dmlc · Aug 21, 2024 · 77c2bb9 · 77c2bb9
1 parent fc4c814
commit 77c2bb9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/graphbolt/pyg/hetero/node_classification.py b/examples/graphbolt/pyg/hetero/node_classification.py
@@ -366,7 +366,7 @@ def parse_args():
     parser.add_argument(
         "--lr",
         type=float,
-        default=0.003,
+        default=0.001,
         help="Learning rate for optimization.",
     )
     parser.add_argument(
@@ -383,9 +383,9 @@ def parse_args():
     parser.add_argument(
         "--fanout",
         type=str,
-        default="10,25",
+        default="25,10",
         help="Fan-out of neighbor sampling. It is IMPORTANT to keep len(fanout)"
-        " identical with the number of layers in your model. Default: ",
+        " identical with the number of layers in your model. Default: 25,10",
     )
     parser.add_argument(
         "--mode",