mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-11-03 11:20:49 +00:00
* model.to("cuda")
model.to("cuda")
* update device placement
---------
Co-authored-by: rasbt <mail@sebastianraschka.com>
Chapter 6: Finetuning for Classification
Main Chapter Code
- 01_main-chapter-code contains the main chapter code
Bonus Materials
- 02_bonus_additional-experiments includes additional experiments (e.g., training the last vs first token, extending the input length, etc.)
- 03_bonus_imdb-classification compares the LLM from chapter 6 with other models on a 50k IMDB movie review sentiment classification dataset