Mingyuan Xu f77c376b05
Run generate example in ch06 optionally on GPU (#352)
* model.to("cuda")

model.to("cuda")

* update device placement

---------

Co-authored-by: rasbt <mail@sebastianraschka.com>
2024-09-13 08:01:52 -05:00
..

Chapter 6: Finetuning for Classification

Main Chapter Code

Bonus Materials