cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix dataset path calculation

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 19:01:39 -04:00
parent 14a302dc5e
commit bbdbe91871
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
training/scripts/train.py
View File

@@ -61,10 +61,10 @@ def train(config_path):
from datasets import load_dataset from datasets import load_dataset
import os import os
# Get dataset paths relative to script location # Get dataset paths - dataset is in training/data/ relative to repo root
script_dir = Path(__file__).parent.parent repo_root = Path(__file__).parent.parent.parent
train_path = str(script_dir / ".." / ".." / "training" / "data" / "train.jsonl") train_path = str(repo_root / "training" / "data" / "train.jsonl")
test_path = str(script_dir / ".." / ".." / "training" / "data" / "test.jsonl") test_path = str(repo_root / "training" / "data" / "test.jsonl")
dataset = load_dataset( dataset = load_dataset(
"json", "json",