NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-28 17:29:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

testing 3

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

2
example_trainer/trainers.py
View file

@ -569,7 +569,7 @@ def train_shared_vllm(config: TrainingConfig):
log_metrics(metrics, step + 1, use_wandb, benchmark=config.benchmark)
# Periodic checkpoint (for recovery, not for vLLM sync)
if (step + 1) % config.vllm_restart_interval == 0:
if config.checkpoint_interval > 0 and (step + 1) % config.checkpoint_interval == 0:
save_checkpoint(model, tokenizer, config.save_path, step + 1)
# === Cleanup ===