bigscience-workshop / bigscience-workshop/bigscience
Zero_Stage=1 results in higher TFLOPS?
- Lenguaje dominante
- Shell
- Estrellas
- 1k
- Forks
- 102
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
### Description
I am learning the chronicles_prequel, and I find the last table in the chapter indicates the higher TFLOPS is achieved with Zero_Stage = 1.
[Trying with ZeRO_STAGE=0/1](https://github.com/bigscience-workshop/bigscience/blob/master/train/tr11-176B-ml/chronicles-prequel.md#48-node-contenders)
Zero_stage=1 could reduce the memory cost, but how come it increases the performance with other parameter being the same?
Nodes | Size | ZS | DP | TP | PP | MBS | GBS | Mem | Sec/it | TFLOPs | Notes
-- | -- | -- | -- | -- | -- | -- | -- | -- | -- | -- | --
48 | 181B | 1 | 4 | 8 | 12 | 2 | 2048 | 37GB | 120.29 | 134.02 | 02-21
48 | 181B | 0 | 4 | 8 | 12 | 2 | 2048 | 72GB | 137.34 | 113.02 | 02-21
Guía de contribución
Línea de trabajo
Comience con la tabla de candidatos de 48 nodos en chronicles-prequel.md, enlazada desde el issue, y compare las filas de Zero Stage 0 y 1 y sus configuraciones registradas. Determine por qué los TFLOPS indicados difieren a pesar de que los parámetros coinciden y, a continuación, aclare la explicación o corrija la documentación para que el resultado sea comprensible.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Área
- documentation, machine-learning, performance
- Tipo de issue
- Documentación
- Dificultad
- 3/5
- Tiempo estimado
- 1-2 días
- Estado de actividad
- Estancado
- Claridad
- Bastante claro
- Aptitud para principiantes
- 35/100