瀏覽代碼

Merge pull request #157 from DrCesar/main

fix move model to device before calculating loss
Sebastian Raschka 1 年之前
父節點
當前提交
738ec44bf9
共有 1 個文件被更改,包括 1 次插入0 次删除
  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()"