add total_training_steps

Author: binary-husky

ajet/default_config/ajet_default.yaml

@@ -243,6 +243,7 @@ ajet:
     # save and test frequency (in step)
     save_freq: 20
     test_freq: 20
+    total_training_steps: null
 
     # total training epochs
     total_epochs: 50

ajet/default_config/verl/config_auto_convertion_verl.jsonc

@@ -35,6 +35,7 @@
     "ajet.lora.target_modules": "actor_rollout_ref.model.target_modules",
     "ajet.lora.load_format": "actor_rollout_ref.rollout.load_format",
 
+    "ajet.trainer_common.total_training_steps": "trainer.total_training_steps",
     "ajet.trainer_common.save_freq": "trainer.save_freq",
     "ajet.trainer_common.test_freq": "trainer.test_freq",