We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d56f80 commit eb1fe22Copy full SHA for eb1fe22
research/gnn-survey-paper/train.py
@@ -16,7 +16,7 @@
16
flags.DEFINE_integer('gpu', '-1', 'Gpu id, -1 means cpu only')
17
flags.DEFINE_float('lr', 1e-2, 'Initial learning rate')
18
flags.DEFINE_integer('epochs', 200, 'Number of training epochs')
19
-flags.DEFINE_integer('num_layers', 2, 'Number of gnn hidden layers')
+flags.DEFINE_integer('num_layers', 2, 'Number of gnn layers')
20
flags.DEFINE_list('hidden_dim', [32], 'Dimension of gnn hidden layers')
21
flags.DEFINE_string('optimizer', 'adam', 'Optimizer for training')
22
flags.DEFINE_float('weight_decay', 5e-4, 'Weight for L2 regularization')
0 commit comments