yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-11-14 17:13:39 +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 9c666a4d94
commit 8c8ff24118
1 changed files with 1 additions and 1 deletions
Show Stats 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]