Browse Source

Merge pull request #157 from DrCesar/main

fix move model to device before calculating loss
Sebastian Raschka 1 year ago
parent
commit
738ec44bf9
1 changed files with 1 additions and 0 deletions
  1. 1 0
      appendix-D/01_main-chapter-code/appendix-D.ipynb

+ 1 - 0
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()"