manual edits

pyg-team · Dec 21, 2024 · 48788cd · 48788cd
1 parent b51e8bb
commit 48788cd
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/examples/llm/g_retriever.py b/examples/llm/g_retriever.py
@@ -185,14 +185,14 @@ def inference_step(model, batch, model_save_name):
 
 
 def train(
-        num_epochs,  # Total number of training epochs
-        hidden_channels,  # Number of hidden channels in GNN
-        num_gnn_layers,  # Number of GNN layers
-        batch_size,  # Training batch size
-        eval_batch_size,  # Evaluation batch size
-        lr,  # Initial learning rate
-        checkpointing=False,  # Whether to checkpoint model
-        tiny_llama=False,  # Whether to use tiny LLaMA model
+    num_epochs,  # Total number of training epochs
+    hidden_channels,  # Number of hidden channels in GNN
+    num_gnn_layers,  # Number of GNN layers
+    batch_size,  # Training batch size
+    eval_batch_size,  # Evaluation batch size
+    lr,  # Initial learning rate
+    checkpointing=False,  # Whether to checkpoint model
+    tiny_llama=False,  # Whether to use tiny LLaMA model
 ):
     """Train a GNN+LLM model on WebQSP dataset.