bigscience-workshop / bigscience-workshop/bigscience

Zero_Stage=1 results in higher TFLOPS?

Ouverte
#48 1 commentaire 0 réactions 0 personnes assignées Voir sur GitHub
Langage dominant
Shell
Étoiles
1k
Forks
102
Métriques de merge des PR
Aucune PR mergée en 30 j

Description

### 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

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Commencez par le tableau des candidats à 48 nœuds dans chronicles-prequel.md, accessible depuis l’issue, et comparez les lignes Zero Stage 0 et 1 ainsi que leurs paramètres enregistrés. Déterminez pourquoi les TFLOPS rapportés diffèrent malgré des paramètres identiques, puis clarifiez l’explication ou corrigez la documentation afin que le résultat soit compréhensible.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Domaine
documentation, machine-learning, performance
Type d'issue
Documentation
Difficulté
3/5
Temps estimé
1-2 jours
Activité
À l'abandon
Clarté
Plutôt claire
Accessibilité débutants
35/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.