Update eval_ogbn_mag.py

acbull · Dec 8, 2020 · d40b377 · d40b377
1 parent afdf3f1
commit d40b377
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ogbn-mag/eval_ogbn_mag.py b/ogbn-mag/eval_ogbn_mag.py
@@ -100,10 +100,11 @@ def prepare_data(pool, task_type = 'train', s_idx = 0, n_batch = args.n_batch, b
             jobs.append(p)
     return jobs
 
+graph = dill.load(open(args.data_dir, 'rb'))
 np.random.seed(43)
 np.random.shuffle(graph.test_paper)
 
-graph = dill.load(open(args.data_dir, 'rb'))
+
 evaluator = Evaluator(name='ogbn-mag')
 device = torch.device("cuda:%d" % args.cuda)
 gnn = GNN(conv_name = args.conv_name, in_dim = len(graph.node_feature['paper'][0]), \