yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-07-28 11:24:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #157 from DrCesar/main

Browse Source 
fix move model to device before calculating loss
This commit is contained in:
Sebastian Raschka 2024-05-15 20:58:43 -04:00 committed by GitHub
commit 738ec44bf9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
appendix-D/01_main-chapter-code/appendix-D.ipynb
View File

@ -425,6 +425,7 @@
"\n", "\n",
"torch.manual_seed(123)\n", "torch.manual_seed(123)\n",
"model = GPTModel(GPT_CONFIG_124M)\n", "model = GPTModel(GPT_CONFIG_124M)\n",
"model.to(device)\n",
"\n", "\n",
"loss = calc_loss_batch(input_batch, target_batch, model, device)\n", "loss = calc_loss_batch(input_batch, target_batch, model, device)\n",
"loss.backward()" "loss.backward()"