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

readme fixes

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-17 13:44:48 -05:00
parent 396491ab72
commit 16ac332880
4 changed files with 63 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

7
example_trainer/api.py
View file

@ -53,6 +53,11 @@ def register_trainer(config: TrainingConfig):
Verifies registration succeeded before returning.
"""
url = config.atropos_url
save_checkpoint_interval = (
config.training_steps
if config.checkpoint_interval <= 0
else config.checkpoint_interval
)
response = requests.post(
f"{url}/register",
json={
@ -63,7 +68,7 @@ def register_trainer(config: TrainingConfig):
"max_token_len": config.seq_len,
"starting_step": 0,
"checkpoint_dir": config.save_path,
"save_checkpoint_interval": config.training_steps,
"save_checkpoint_interval": save_checkpoint_interval,
"num_steps": config.training_steps,
},
timeout=10,