From 6f17e5b32a0d0ca7ccaca78232f9d4bb249879e1 Mon Sep 17 00:00:00 2001
From: Shiqiang Zhang <shiqiang.imperial@gmail.com>
Date: Thu, 18 Apr 2024 09:03:07 +0100
Subject: [PATCH] Update model_training_banana.py

---
 model_training_banana.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_training_banana.py b/model_training_banana.py
index 01cf360..0e55206 100644
--- 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()