fix(jax): honor training.stat_file for model statistics
Los mantenedores suelen responder en 4 días
Nadie ha tomado este issue todavía.
Evaluación
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Aptitud para principiantes
- 48/100
- Tipo de issue
- Error
- Claridad
- Bastante claro
- Estado de actividad
- Tranquilo
- Stack tecnológico
- python
Línea de trabajo
Comienza en el punto de entrada del entrenamiento de JAX y sigue cómo training.stat_file llega a la carga o al cálculo de estadísticas. Verifica las comprobaciones de compatibilidad existentes, la sincronización distribuida y la gestión de recursos; después, usa los criterios del issue sobre carga, creación, incompatibilidad y propiedad distribuida para definir los criterios de finalización y añadir las pruebas solicitadas.
Escrito por el modelo de indexación a partir del texto del issue.
Descripción
Summary
JAX training carries training.stat_file through configuration, but the entrypoint currently invokes statistics computation/loading without passing the configured file path. The option is therefore accepted but appears to be ignored.
Scope
- Resolve the configured statistics path consistently with the training working directory.
- Pass it into JAX model statistics loading/computation.
- Reuse existing statistics when compatible and write newly computed statistics safely.
- Validate model/type-map/descriptor compatibility before accepting a file.
- Define chief-only file I/O and synchronization for distributed training.
- Close HDF5/file resources on success and failure.
Acceptance criteria
- A configured compatible statistics file is loaded instead of recomputing statistics.
- A missing target file is created at the configured location when writing is allowed.
- Incompatible or malformed files produce actionable errors.
- Only the designated rank performs mutable file I/O in distributed runs, and all ranks receive identical statistics.
- Tests cover load, create, incompatibility, and distributed ownership paths.
Refs #5756.
Coding agent: Codex
Codex version: codex-cli 0.144.4
Model: gpt-5.6-sol
Reasoning effort: xhigh
- Lenguaje dominante
- Python
- Estrellas
- 2.1k
- Forks
- 659
- Merge medio
- 4 d 2 h
- PR fusionados (30 d)
- 6
Preparar el entorno
Inicia el contenedor de desarrollo del proyecto en tu navegador, con tu propia cuenta de GitHub.
- Sin Dockerfile ni archivo de Docker Compose
- Sin plantilla de pull request
- Leer la guía de contribución
Primeros pasos
- Lee el issue completo y luego la guía de contribución del proyecto.
- Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
- Haz un fork del repositorio y trabaja en una rama.
- Abre un pull request que haga referencia al número del issue.
Más de deepmodeling/deepmd-kit
-
Dificultad 1/5 Menos de una hora Aptitud para principiantes 88/100
deepmodeling/deepmd-kit#6038 · 2 comentarios ·
Los mantenedores suelen responder en 4 días
-
[BUG] Native-spin DPA4C graph path fails on zero-node graphsPosiblemente ocupada Un pull request vinculado a esta issue está abierto o ya se fusionó. Abiertobug
Dificultad 2/5 1-3 horas Aptitud para principiantes 78/100
deepmodeling/deepmd-kit#5991 ·
Los mantenedores suelen responder en 4 días
-
Docs enhancement
Dificultad 2/5 1-3 horas Aptitud para principiantes 68/100
deepmodeling/deepmd-kit#5766 · 1 comentario ·
Los mantenedores suelen responder en 4 días
-
[Code scan] Symmetrize ASE calculator stress after reshaping virialPosiblemente ocupada @njzjz la tomó hace 93 días. Abiertobug
Dificultad 2/5 1-3 horas Aptitud para principiantes 72/100
deepmodeling/deepmd-kit#5689 · 2 comentarios ·
Los mantenedores suelen responder en 4 días
-
[Code scan] Freeze Paddle DPA3 parameters with stop_gradientPosiblemente ocupada @njzjz la tomó hace 93 días. Abiertobug
Dificultad 2/5 1-3 horas Aptitud para principiantes 78/100
deepmodeling/deepmd-kit#5686 · 1 comentario ·
Los mantenedores suelen responder en 4 días
Todos los issues de deepmodeling/deepmd-kit
Issues similares
-
mermaid_extract.py silently discards single-line state-diagram notes; sequence notes are keptAbierto
Dificultad 2/5 1-3 horas Aptitud para principiantes 82/100
cathrynlavery/diagram-design#441 ·
Los mantenedores suelen responder en 2 días
-
[Cleanup] Include the public header in work_split.cppPosiblemente ocupada @blozano-tt la tomó hoy. Abierto
Dificultad 2/5 1-3 horas Aptitud para principiantes 82/100
tenstorrent/tt-metal#60850 ·
Los mantenedores suelen responder en 1 día
-
Dificultad 2/5 1-3 horas Aptitud para principiantes 78/100
Los mantenedores suelen responder en 1 día
-
backend::sglang bug language::python observability
Dificultad 2/5 Medio día Aptitud para principiantes 65/100
Los mantenedores suelen responder en 1 día
-
Dificultad 1/5 Menos de una hora Aptitud para principiantes 75/100
OpenShot/openshot-qt#6202 ·
Los mantenedores suelen responder en 1 día