remove redundant line

timmyjose-forks · Jun 20, 2024 · 6c84900 · 6c84900
1 parent 977f9c6
commit 6c84900
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/ch05/01_main-chapter-code/exercise-solutions.ipynb b/ch05/01_main-chapter-code/exercise-solutions.ipynb
@@ -431,7 +431,6 @@
     "\n",
     "optimizer = torch.optim.AdamW(model.parameters(), lr=0.0004, weight_decay=0.1)\n",
     "optimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\n",
-    "model.to(device)\n",
     "model.train();"
    ]
   },