mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-06 11:37:44 +00:00
feat: advanced checkpoint selectors (#790)
* rebase selection code * bug-free run: checkpoint selection and dynamic EDA loading * add prototypes of various selectors, to imp. and test later * fix EDA write bug * imp SOTA-Jump policy * fix small bug * allow to set different selector by .env * add always-win selector * add init length for AlwaysWinCKPSelector * add back_jump selector * auto lint * add sota_exp_to_submit attribute; change the name of ckp_selector and sota-selector * fix bug * auto lint * working on auto sota selector * add subtrace counter * fix bug, remove unuse selector * add auto sota selector * auto lint * fix bug * fix small logic bug * add logging * add inject_diverse feat * auto lint * capable to None-select * feat: add hypothesis_gen config and ExpGen2TraceAndMerge functionality * refactor: use dynamic import for experiment generator instantiation * feat: add BestValidSelector for improved SOTA experiment selection * runnable twin-trace version * fix logic error of trace-merge * auto lint * use import_class to set selector, * auto-lint --------- Co-authored-by: Young <afe.young@gmail.com>
This commit is contained in:
@@ -149,11 +149,22 @@ class CheckpointSelector:
|
||||
- `(idx, )` represents starting from the `idx`-th trial in the trace.
|
||||
- `None` represents starting from scratch (start a new trace)
|
||||
|
||||
|
||||
- More advanced selection strategies in `select.py`
|
||||
"""
|
||||
|
||||
|
||||
class SOTAexpSelector:
|
||||
"""
|
||||
Select the SOTA experiment from the trace to submit
|
||||
"""
|
||||
|
||||
@abstractmethod
|
||||
def get_sota_exp_to_submit(self, trace: Trace) -> Experiment | None:
|
||||
"""
|
||||
Select the SOTA experiment from the trace to submit
|
||||
"""
|
||||
|
||||
|
||||
class ExpGen(ABC):
|
||||
|
||||
def __init__(self, scen: Scenario) -> None:
|
||||
|
||||
Reference in New Issue
Block a user