Add Auto-FL literature events and local compute budget search#4523
Add Auto-FL literature events and local compute budget search#4523holgerroth merged 20 commits intoNVIDIA:mainfrom
Conversation
|
/build |
Greptile SummaryThis PR extends the Auto-FL research harness with three related capabilities: literature-review event tracking (new
Confidence Score: 5/5Safe to merge — all changed code paths are well-bounded, and no training or scoring logic is altered in a way that would corrupt existing results. The step-based training path is properly isolated behind local_train_steps > 0, with matching validation in both job.py and client.py, correct NUM_STEPS_CURRENT_ROUND bookkeeping, and consistent scheduler and scaffold handling. The new literature-event scripts are append-only helpers that do not touch training outcomes. The only non-trivial concern is a cosmetic ambiguity in the repeated_terms diagnostic printed by plateau_watchdog.py, which does not affect the recommendation output the automation consumes. No files require special attention beyond a close read of the plateau_watchdog.py diagnostic output logic. Important Files Changed
Reviews (7): Last reviewed commit: "Merge branch 'main' into codex/auto-fl-l..." | Re-trigger Greptile |
|
/build |
ZiyueXu77
left a comment
There was a problem hiding this comment.
nice to have literature and plateau watch
|
/build |
|
/build |
1 similar comment
|
/build |
|
/build |
|
/build |
Summary
--local_train_stepsalongside epoch-based trainingValidation
python3 -m py_compile research/auto-fl-research/client.py research/auto-fl-research/job.py research/auto-fl-research/scripts/append_result.py research/auto-fl-research/scripts/validate_contract.pyblack --check research/auto-fl-research/client.py research/auto-fl-research/job.pyisort --check-only --profile black research/auto-fl-research/client.py research/auto-fl-research/job.pyflake8 research/auto-fl-researchgit diff --checkmake validatefromresearch/auto-fl-researchmake smokefromresearch/auto-fl-researchpassed static checks and skipped runtime launch because the host Python has incompatible NVFlare API paths installedNotes
results.tsvare not included in this PR.