Add hyperparameter search toolkit and Modal GPU training script#2

Merged

williamQyq merged 11 commits into

williamQyq:mainfrom

Northeastern-Research-ORNL-1:main

Apr 6, 2026

Commits on Feb 26, 2026

optuna, w&b, modal scripts
KrishnaRaheja
committed

Commits on Mar 9, 2026

[Fix] update EPOCHS_PER_TRIAL and dropout range for hyperparameter search
KrishnaRaheja
committed

Commits on Mar 15, 2026

Add script to train final model on Modal GPU with best Optuna parameters
KrishnaRaheja
committed
reflectivity_pipeline_experimental is a run of 6-layer cnn model with 6 layers, 0.107 dropout, and 64 batch size.
KrishnaRaheja
committed

Commits on Mar 17, 2026

reflectivity_pipeline_experimental_125k
KrishnaRaheja
committed

Commits on Apr 1, 2026

Fix two bugs in train_modal.py caught by code review
KrishnaRaheja
committed

Commits on Apr 3, 2026

Add experiment setup docs for data generation before Modal training
KrishnaRaheja
committed