training time

pyg-team · Oct 1, 2024 · 6846130 · 6846130
1 parent b160381
commit 6846130
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/examples/llm/ogbg_code2.py b/examples/llm/ogbg_code2.py
@@ -13,7 +13,6 @@
 def get_loss_ogbg(model, batch, **kwargs) -> torch.Tensor:
     questions = [master_prompt for i in range(len(batch.y))]
     labels = ['|'.join(label) for label in batch.y]
-    print("batch.desc=", batch.desc)
     return model(questions, batch.x.to(torch.float), batch.edge_index,
                  batch.batch, labels, batch.edge_attr, batch.desc)