Is current megablocks compatible with distributed optimizer in Megatron-LM?

Abierto
#160 1 comentario 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
25/100
Tipo de issue
Error
Claridad
Necesita aclaración
Estado de actividad
Estancado
Stack tecnológico
python

Línea de trabajo

Comienza con las líneas 352-356 de megatron/arguments.py en el fork enlazado y compara después esa validación con NVIDIA Megatron-LM pull request #288. Rastrea cómo interactúa el paralelismo de expertos con el optimizador distribuido y documenta si la aserción es necesaria, idealmente con una incompatibilidad reproducida o una confirmación de compatibilidad.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

Hi there, thanks for the amazing work! I found expert parallel is not compatible with the distributed optimizer in the fork version of Megatron-LM here:

https://github.com/stanford-futuredata/Megatron-LM/blob/85f95aef3b648075fe6f291c86714fdcbd9cd1f5/megatron/arguments.py#L352-L356

But there's no such validation in the open PR to Megatron-LM: https://github.com/NVIDIA/Megatron-LM/pull/288

Does that mean the assertion is redundant and the current version of megablocks is compatible with the distributed optimizer under expert parallelism?

Thanks very much.

Lenguaje dominante
Python
Estrellas
1.6k
Forks
227
Métricas de merge de PR
Sin PR fusionados en 30 d

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.