tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-07-08 13:59:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 05c8ab093b into 0aaca56805

Browse Source
This commit is contained in:
Suraj-Self 2026-04-19 12:41:16 +08:00 committed by GitHub
commit e11bc43aea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nanochat/core_eval.py
View File

@ -177,7 +177,7 @@ def evaluate_example(idx, model, tokenizer, data, device, task_meta):
if num_fewshot > 0: if num_fewshot > 0:
rng = random.Random(1234 + idx) rng = random.Random(1234 + idx)
available_indices = [i for i in range(len(data)) if i != idx] available_indices = [i for i in range(len(data)) if i != idx]
fewshot_indices = rng.sample(available_indices, num_fewshot) fewshot_indices = rng.sample(available_indices, min(num_fewshot, len(available_indices)))
fewshot_examples = [data[i] for i in fewshot_indices] fewshot_examples = [data[i] for i in fewshot_indices]
# Render prompts and batch sequences based on task type # Render prompts and batch sequences based on task type