training time

pyg-team · Oct 1, 2024 · 0fa3490 · 0fa3490
1 parent 5d9d16f
commit 0fa3490
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/llm/g_retriever.py b/examples/llm/g_retriever.py
@@ -243,7 +243,7 @@ def adjust_learning_rate(param_group, LR, epoch):
             print("testing on 500 random instances to start...")
             break
         with torch.no_grad():
-            eval_output.append(inference_step(model, batch, model_save_name))
+            eval_output.append(inference_step(model, batch, model_save_name=model_save_name))
         progress_bar_test.update(1)
 
     compute_metrics(eval_output)