Update model_training_banana.py

cog-imperial · Apr 18, 2024 · 6f17e5b · 6f17e5b
1 parent 1efecbb
commit 6f17e5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_training_banana.py b/model_training_banana.py
@@ -26,7 +26,7 @@ def get(self, idx):
 
 print(f"Number of node features: {dataset.num_node_features}")
 
-seed_gnn = 0
+seed_gnn = int(sys.argv[1])
 
 torch.manual_seed(seed_gnn)
 dataset_shuffle = dataset.shuffle()