fix max steps

kumo-ai · Oct 21, 2024 · 97bacbd · 97bacbd
1 parent 9346b61
commit 97bacbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ijcai_example.py b/examples/ijcai_example.py
@@ -39,7 +39,7 @@
 parser.add_argument("--num_layers", type=int, default=2)
 parser.add_argument("--num_neighbors", type=int, default=128)
 parser.add_argument("--temporal_strategy", type=str, default="last")
-parser.add_argument("--max_steps_per_epoch", type=int, default=2)
+parser.add_argument("--max_steps_per_epoch", type=int, default=2000)
 parser.add_argument("--num_workers", type=int, default=0)
 parser.add_argument("--eval_k", type=int, default=10)
 parser.add_argument("--seed", type=int, default=42)