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

Test code in pytorch 2.4 (#285)

Browse Source 
* test code in pytorch 2.4

* update
This commit is contained in:
Sebastian Raschka
2024-07-24 21:53:41 -05:00
committed by GitHub
parent 6cbe6520a2
commit 08040f024c
17 changed files with 509 additions and 377 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ch05/01_main-chapter-code/gpt_train.py
View File

@@ -239,4 +239,4 @@ if __name__ == "__main__":
# Save and load model
torch.save(model.state_dict(), "model.pth")
model = GPTModel(GPT_CONFIG_124M)
model.load_state_dict(torch.load("model.pth"))
model.load_state_dict(torch.load("model.pth"), weights_only=True)