NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

testing 3

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-02 13:08:42 -05:00
parent dd7047dd5a
commit 8a9e6945ee
4 changed files with 18 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
example_trainer/cli.py
View file

@ -82,6 +82,12 @@ def parse_args() -> argparse.Namespace:
default="trained_model_checkpoints",
help="Directory to save model checkpoints",
)
parser.add_argument(
"--checkpoint-interval",
type=int,
default=3,
help="Save checkpoint every N training steps (0 = only save final)",
)
# === vLLM Arguments ===
parser.add_argument(
@ -258,6 +264,7 @@ def config_from_args(args: argparse.Namespace) -> TrainingConfig:
optimizer=args.optimizer,
device=args.device,
save_path=args.save_path,
checkpoint_interval=getattr(args, "checkpoint_interval", 3),
vllm_restart_interval=args.vllm_restart_interval,
vllm_port=args.vllm_port,
vllm_gpu_memory_utilization=args.vllm_gpu_memory_utilization,