From a12bac949631b75473aa098ebe6178ace6e53d67 Mon Sep 17 00:00:00 2001 From: zhanke Date: Fri, 18 Sep 2020 10:42:19 +0800 Subject: [PATCH] change seed for dropout --- model_zoo/official/gnn/gcn/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/gnn/gcn/train.py b/model_zoo/official/gnn/gcn/train.py index f7b0c600987..f31ce441f99 100644 --- a/model_zoo/official/gnn/gcn/train.py +++ b/model_zoo/official/gnn/gcn/train.py @@ -49,7 +49,7 @@ def train(): """Train model.""" parser = argparse.ArgumentParser(description='GCN') parser.add_argument('--data_dir', type=str, default='./data/cora/cora_mr', help='Dataset directory') - parser.add_argument('--seed', type=int, default=123, help='Random seed') + parser.add_argument('--seed', type=int, default=0, help='Random seed') parser.add_argument('--train_nodes_num', type=int, default=140, help='Nodes numbers for training') parser.add_argument('--eval_nodes_num', type=int, default=500, help='Nodes numbers for evaluation') parser.add_argument('--test_nodes_num', type=int, default=1000, help='Nodes numbers for test')