div18 commited on
Commit
85b91e2
·
1 Parent(s): ad95d2f

show script

Browse files
Files changed (1) hide show
  1. training/train.py +11 -6
training/train.py CHANGED
@@ -16,14 +16,19 @@ Architecture (from training.md):
16
  - Metrics = structured + queryable
17
 
18
  Usage:
19
- # On HF Jobs (A10G recommended):
20
- python training/train.py
21
 
22
- # Local with custom config:
23
- HF_SPACE_URL=http://localhost:8000 python training/train.py --config my_config.yaml
 
 
24
 
25
- # Override specific values via CLI:
26
- python training/train.py --num-iterations 100 --loss-type grpo
 
 
 
27
  """
28
 
29
  from __future__ import annotations
 
16
  - Metrics = structured + queryable
17
 
18
  Usage:
19
+ # RECOMMENDED: Launch via HF Jobs (auto-provisions GPU, pushes to Hub):
20
+ python training/launch_train.py --run-id my_run
21
 
22
+ # Or run directly (requires a running AntiAtropos server):
23
+ ANTIATROPOS_ENV_URL=http://localhost:8000 \
24
+ ANTIATROPOS_HUB_MODEL_REPO=Keshav051/antiatropos-qlora \
25
+ python training/train.py --run-id my_run --num-iterations 15
26
 
27
+ # Override defaults:
28
+ python training/train.py --run-id run_007 --num-iterations 500 --num-episodes 6
29
+
30
+ # See all options:
31
+ python training/train.py --help
32
  """
33
 
34
  from __future__ import annotations