yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-10-31 01:41:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix move model to device before calculating loss

Browse Source
This commit is contained in:
DrCesar 2024-05-14 22:28:00 -07:00
parent e4227c899b
commit d2410cb0c6
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()"