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

more terminal changes

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-02 14:40:55 -05:00
parent 2f01720899
commit 8d29f49a58
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
example_trainer/training.py
View file

@ -287,7 +287,7 @@ def run_training_step(
temperature_batches: List of temperature tensors
config: Training configuration (includes clip_eps, warmup_steps)
step_idx: Current global training step (0-based)
inference_logprob_batches: Batched logprobs from inference (π_old), aligned with labels
inference_logprob_batches: Rollout logprobs from inference, aligned with labels
Returns:
Dict of training metrics for this step