소스 검색

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()"