books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use test mode arg in ch07 (#713)

Browse Source
This commit is contained in:
Sebastian Raschka
2025-06-27 19:28:56 -05:00
committed by GitHub
parent 8b3e4b24b0
commit ddbaf0d83e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ch07/01_main-chapter-code/gpt_instruction_finetuning.py
View File

@@ -175,7 +175,7 @@ def main(test_mode=False):
val_data = data[train_portion + test_portion:] val_data = data[train_portion + test_portion:]
# Use very small subset for testing purposes # Use very small subset for testing purposes
if args.test_mode: if test_mode:
train_data = train_data[:10] train_data = train_data[:10]
val_data = val_data[:10] val_data = val_data[:10]
test_data = test_data[:10] test_data = test_data[:10]