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

remove training code

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-13 12:52:52 -04:00
parent 862cd3667d
commit 148a4fd5eb
6 changed files with 38 additions and 329 deletions
Download patch file Download diff file Expand all files Collapse all files

3
example_trainer/run.py
View file

@ -201,9 +201,6 @@ def main():
checkpoint_interval=args.checkpoint_interval,
# GRPO hyperparameters
clip_eps=args.clip_eps,
distill_enabled=getattr(args, "distill_enabled", False),
distill_coef=getattr(args, "distill_coef", 0.0),
distill_temperature=getattr(args, "distill_temperature", 1.0),
# vLLM settings
vllm_port=args.vllm_port,
vllm_gpu_memory_utilization=args.gpu_memory_utilization,