yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-08-29 11:00:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove redundant line

Browse Source
This commit is contained in:
rasbt 2024-06-20 10:12:28 -05:00
parent cb194fa8fa
commit e1046746e8
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ch05/01_main-chapter-code/exercise-solutions.ipynb
View File

@ -431,7 +431,6 @@
"\n", "\n",
"optimizer = torch.optim.AdamW(model.parameters(), lr=0.0004, weight_decay=0.1)\n", "optimizer = torch.optim.AdamW(model.parameters(), lr=0.0004, weight_decay=0.1)\n",
"optimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\n", "optimizer.load_state_dict(checkpoint[\"optimizer_state_dict\"])\n",
"model.to(device)\n",
"model.train();" "model.train();"
] ]
}, },