From ecd2855334dfdaaca2a69ed208b6b76dd8d4a540 Mon Sep 17 00:00:00 2001
From: DrCesar <developer@drcesar.tech>
Date: Tue, 14 May 2024 22:28:00 -0700
Subject: [PATCH] fix move model to device before calculating loss

---
 appendix-D/01_main-chapter-code/appendix-D.ipynb | 1 +
 1 file changed, 1 insertion(+)

diff --git a/appendix-D/01_main-chapter-code/appendix-D.ipynb b/appendix-D/01_main-chapter-code/appendix-D.ipynb
index 435c6c2..993bfb8 100644
--- 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()"