Update train.py

train.py

@@ -47,7 +47,7 @@ class HiMoEConfig:
     num_experts:  int   = 8    # Level-2 choices per MoE
     # Training
     batch_size:   int   = 32
-    max_iters:    int   = 50 # for testing, increase to 3000 for actual training
+    max_iters:    int   = 3000
     eval_interval:int   = 50
     eval_iters:   int   = 20
     lr:           float = 3e-4