Feature/hp opt - #4

Merged

knc6 merged 15 commits into

atomgptlab:mainfrom

crhysc:feature/hp_opt

Sep 24, 2025

Commits on Aug 28, 2025

add hyperparameter_search.py
crhysc
committed
6d27eed
add an example hp_search_config.json
Charles Campbell
committed
850af0b
add a field that points to the hp_search_config.json
Charles Campbell
committed
b26541d
add optuna>=3.5
Charles Campbell
committed
4e36a22
fix typo and replace best_values with best_trials
crhysc
committed
258a75f
optuna>=3.5,<4
Charles Campbell
committed
43722fe
import TrainingPropConfig in hp_search.py instead of making a new one
crhysc
committed
3013988
send SFTTrainer hardcoded args to config and add hp_opt training config parameters to config
crhysc
committed
287d7c0
use strings for eval and save strategies
crhysc
committed
a04c0ad
remove customsfttrainer and use regular hf sfttrainer
crhysc
committed
5bb5daf
try-except-pass wrap for del m._flag_for_generation
crhysc
committed
8595698
replace tokenizer= with processing_class=
Charles Campbell
committed
9eca98b
remove dataset_text_field
Charles Campbell
committed
d0be0a6
add sft_args to conform to atomgpt_compiled_cache
crhysc
committed
b3e5ac0
update to a more robust example
Charles Campbell
committed
3b3ed56