Skip to content

Commit eb1fe22

Browse files
committed
Fix typo
1 parent 9d56f80 commit eb1fe22

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/gnn-survey-paper/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
flags.DEFINE_integer('gpu', '-1', 'Gpu id, -1 means cpu only')
1717
flags.DEFINE_float('lr', 1e-2, 'Initial learning rate')
1818
flags.DEFINE_integer('epochs', 200, 'Number of training epochs')
19-
flags.DEFINE_integer('num_layers', 2, 'Number of gnn hidden layers')
19+
flags.DEFINE_integer('num_layers', 2, 'Number of gnn layers')
2020
flags.DEFINE_list('hidden_dim', [32], 'Dimension of gnn hidden layers')
2121
flags.DEFINE_string('optimizer', 'adam', 'Optimizer for training')
2222
flags.DEFINE_float('weight_decay', 5e-4, 'Weight for L2 regularization')

0 commit comments

Comments
 (0)