fix #4967

pyg-team · Jul 12, 2022 · 38befa3 · 38befa3
1 parent 845e5d8
commit 38befa3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmark/citation/gcn.py b/benchmark/citation/gcn.py
@@ -38,7 +38,7 @@ def forward(self, data):
         return F.log_softmax(x, dim=1)
 
 
-dataset = get_planetoid_dataset(args.dataset, args.no_normalize_features)
+dataset = get_planetoid_dataset(args.dataset, not args.no_normalize_features)
 permute_masks = random_planetoid_splits if args.random_splits else None
 run(dataset, Net(dataset), args.runs, args.epochs, args.lr, args.weight_decay,
     args.early_stopping, permute_masks)
diff --git a/benchmark/citation/sgc.py b/benchmark/citation/sgc.py
@@ -34,7 +34,7 @@ def forward(self, data):
         return F.log_softmax(x, dim=1)
 
 
-dataset = get_planetoid_dataset(args.dataset, args.no_normalize_features)
+dataset = get_planetoid_dataset(args.dataset, not args.no_normalize_features)
 permute_masks = random_planetoid_splits if args.random_splits else None
 run(dataset, Net(dataset), args.runs, args.epochs, args.lr, args.weight_decay,
     args.early_stopping, permute_masks)