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

rename hparams to settings

Browse Source
This commit is contained in:
rasbt
2024-04-05 07:24:46 -05:00
parent 7d0b9b78b0
commit c31e99720d
6 changed files with 39 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ch05/01_main-chapter-code/tests.py
View File

@@ -23,7 +23,7 @@ def gpt_config():
@pytest.fixture
def other_hparams():
def other_settings():
return {
"learning_rate": 5e-4,
"num_epochs": 1, # small for testing efficiency
@@ -32,8 +32,8 @@ def other_hparams():
}
def test_main(gpt_config, other_hparams):
train_losses, val_losses, tokens_seen, model = main(gpt_config, other_hparams)
def test_main(gpt_config, other_settings):
train_losses, val_losses, tokens_seen, model = main(gpt_config, other_settings)
assert len(train_losses) == 39, "Unexpected number of training losses"
assert len(val_losses) == 39, "Unexpected number of validation losses"