LLMs-from-scratch/ch05/05_bonus_hparam_tuning
Daniel Kleine bbb2a0c3d5
fixed num_workers (#229)
* fixed num_workers

* ch06 & ch07: added num_workers to create_dataloader_v1
2024-06-19 17:36:46 -05:00
..
2024-06-11 20:59:48 -05:00
2024-06-19 17:36:46 -05:00
2024-04-10 22:09:46 -04:00

Optimizing Hyperparameters for Pretraining

The hparam_search.py script, based on the extended training function in Appendix D: Adding Bells and Whistles to the Training Loop, is designed to find optimal hyperparameters via grid search.

Note

This script will take a long time to run. You may want to reduce the number of hyperparameter configurations explored in the HPARAM_GRID dictionary at the top.