feat(tf2): align optimizer dispatch with the common schema
I maintainer di solito rispondono entro 4 giorni
Nessuno ha ancora preso questa issue.
Valutazione
- Difficoltà
- 5/5
- Tempo stimato
- Più di una settimana
- Idoneità per principianti
- 35/100
- Tipo di issue
- Funzionalità
- Chiarezza
- Abbastanza chiara
- Stato di attività
- Tranquilla
- Stack tecnologico
- python, tensorflow
- Ambito
- backend, machine-learning, testing
Direzione di ricerca
Inizia dal trainer TF2 e dal confine condiviso dispatch/factory, quindi traccia il percorso con cui il blocco optimizer normalizzato di primo livello raggiunge la validazione del backend. Esegui o aggiungi test deterministici di aggiornamento a un solo passo per i parametri supportati e verifica che checkpoint/resume riproduca l’aggiornamento successivo. Il lavoro è completato quando ogni schema optimizer ha uno stato TF2 esplicito, le scelte non supportate falliscono prima dell’addestramento e HybridMuon può essere integrato tramite #5771 senza parsing duplicato.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Descrizione
Summary
TF2 optimizer dispatch supports Adam and AdamW, while the common schema also exposes LKF, AdaMuon, and HybridMuon. Unsupported choices currently fail only in the TF2 trainer.
DPA4 HybridMuon support is tracked in #5771; this issue covers the broader backend-aware optimizer contract, including explicit support/restriction for the remaining schema variants.
Scope
- Consume and validate the normalized top-level
optimizerblock. - Implement TF2-native optimizers that are intended to be supported.
- Reject intentionally unsupported variants during backend-aware validation.
- Wire optimizer-specific parameters and preserve state through checkpoints.
- Do not silently fall back to Adam or ignore normalized fields.
- Coordinate the shared dispatch/factory boundary with #5771.
Acceptance criteria
- Every schema-exposed optimizer has an explicit TF2 support status.
- Supported optimizer parameters affect deterministic one-step update tests.
- Unsupported variants fail before the training loop with an actionable message.
- Adam and AdamW remain behaviorally compatible.
- Checkpoint/resume restores complete optimizer state and reproduces the next update.
- HybridMuon integration can land through #5771 without duplicating configuration parsing.
Refs #5757.
Coding agent: Codex
Codex version: codex-cli 0.144.4
Model: gpt-5.6-sol
Reasoning effort: xhigh
- Lingua principale
- Python
- Stelle
- 2.1k
- Fork
- 659
- Merge medio
- 4g 2h
- PR unite (30g)
- 6
Preparare l'ambiente
Avvia il container di sviluppo del progetto nel browser, con il tuo account GitHub.
- Nessun Dockerfile né file Docker Compose
- Nessun modello di pull request
- Leggi la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di deepmodeling/deepmd-kit
-
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 88/100
deepmodeling/deepmd-kit#6038 · 2 commenti ·
I maintainer di solito rispondono entro 4 giorni
-
[BUG] Native-spin DPA4C graph path fails on zero-node graphsForse già presa Una pull request collegata a questa issue è aperta o già unita. Apertabug
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
deepmodeling/deepmd-kit#5991 ·
I maintainer di solito rispondono entro 4 giorni
-
Docs enhancement
Difficoltà 2/5 1-3 ore Idoneità per principianti 68/100
deepmodeling/deepmd-kit#5766 · 1 commento ·
I maintainer di solito rispondono entro 4 giorni
-
[Code scan] Symmetrize ASE calculator stress after reshaping virialForse già presa @njzjz l’ha presa 93 giorni fa. Apertabug
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
deepmodeling/deepmd-kit#5689 · 2 commenti ·
I maintainer di solito rispondono entro 4 giorni
-
[Code scan] Freeze Paddle DPA3 parameters with stop_gradientForse già presa @njzjz l’ha presa 92 giorni fa. Apertabug
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
deepmodeling/deepmd-kit#5686 · 1 commento ·
I maintainer di solito rispondono entro 4 giorni
Tutte le issue di deepmodeling/deepmd-kit
Issue simili
-
`burr` fails to start when the Python path has a space in itForse già presa @enekoreto l’ha presa oggi. Apertastatus/needs-triage
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
I maintainer di solito rispondono entro 5 giorni
-
gemini 出现了新的域名需要加入Aperta
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 72/100
-
Difficoltà 2/5 Mezza giornata Idoneità per principianti 72/100
debpalash/VoiceStudio#2903 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
dagster-io/community-integrations#367 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 74/100
aws-samples/sample-per-model-bedrock#243 ·
I maintainer di solito rispondono entro 1 giorno