Update train.py

train.py

@@ -44,10 +44,11 @@ except Exception as e:
     print(f"Repo create: {e}")
 
 MAX_SEQ_LEN = 1024
-SFT_STEPS = 40  # More warmup for JSON format
+SFT_STEPS = 80
+# More warmup for JSON format
 GRPO_STEPS = 250
 GRPO_K = 2
-GRPO_LR = 1e-5
+GRPO_LR = 2e-5
 CURRICULUM_SWITCH = 0  # Start with Level 1, advance early
 GRAD_CLIP = 1.0
 SAVE_EVERY = 50