Initial commit
Mmaya committed 15 days ago · maincommit 050ce7874acfb564e798213450b87d402a66c98a
Agent sessions behind this commit0
No agent session linked. A person made this commit by hand.
Showing 5 changed files with 26 additions and 0 deletions.
.gitignore+3 −0
@@ -0,0 +1,3 @@
+logs/
+runs/
+reports/
.trail.yaml+4 −0
@@ -0,0 +1,4 @@
+store: supabase://trail/main
+repo: ml-pipeline
+harnesses: [claude, codex]
+artifacts: ["logs/", "runs/**/metrics.json"]
README.md+3 −0
@@ -0,0 +1,3 @@
+# ml-pipeline
+
+Fine-tuning for the support assistant. LoRA on a 7B base model.
data.py+10 −0
@@ -0,0 +1,10 @@
+import json
+
+
+def split(prompts, eval_frac=0.05):
+ seen = set()
+ for p in prompts:
+ key = p
+ if key in seen:
+ continue
+ seen.add(key)
train.py+6 −0
@@ -0,0 +1,6 @@
+MODEL = "llama-7b"
+LORA_RANK = 64
+LR = 3e-4
+WARMUP_STEPS = 100
+GRAD_CLIP = None
+EARLY_STOP_PATIENCE = None