bigscience-workshop / bigscience-workshop/bigscience

Zero_Stage=1 results in higher TFLOPS?

Abierto
#48 1 comentario 0 reacciones 0 asignados Ver en GitHub
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

Abrir la 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

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.