fix move model to device before calculating loss

2025-08-31 03:50:23 +00:00 · 2024-05-14 22:28:00 -07:00 · 2024-05-14 22:28:00 -07:00 · d2410cb0c6
commit d2410cb0c6
parent e4227c899b
1 changed files with 1 additions and 0 deletions
--- a/appendix-D/01_main-chapter-code/appendix-D.ipynb
+++ b/appendix-D/01_main-chapter-code/appendix-D.ipynb
@ -425,6 +425,7 @@
    "\n",
    "torch.manual_seed(123)\n",
    "model = GPTModel(GPT_CONFIG_124M)\n",
+    "model.to(device)\n",
    "\n",
    "loss = calc_loss_batch(input_batch, target_batch, model, device)\n",
    "loss.backward()"