6 lines
488 B
Markdown
Raw Normal View History

2024-03-23 07:27:43 -05:00
# Optimizing Hyperparameters for Pretraining
The [hparam_search.py](hparam_search.py) script, based on the extended training function in [Appendix D: Adding Bells and Whistles to the Training Loop](../../appendix-D/01_main-chapter-code/appendix-D.ipynb), is designed to find optimal hyperparameters via grid search.
2024-03-23 07:27:43 -05:00
>[!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.