Integrating Ray for Hyperparameter Tuning#10

Open

xpkcs wants to merge 11 commits intomainfrom

Commits on May 6, 2025

copying train_clt_local.py so we can see git diff easier
Andy Kim
committed
adding ray tune changes
Andy Kim
committed
adding option to CLTTrainer for ray reporting for use with ray train/tune
Andy Kim
committed

Commits on May 7, 2025

passing configs to worker
Andy Kim
committed
moved configs into train_loop_per_worker func
Andy Kim
committed
updated model checkpointing to work with ray
Andy Kim
committed
created separate trainer class for Ray
Andy Kim
committed
restoring non-ray clt trainer code back to before ray changes
Andy Kim
committed
making use_ray param required
Andy Kim
committed
explicitly setting cpu per worker and example hps
Andy Kim
committed
cleaning up init func
Andy Kim
committed