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

オープン
#160 コメント 1 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

評価

難易度
4/5
見積もり時間
3〜5日
初心者へのやさしさ
25/100
issue の種類
バグ
明瞭さ
説明が足りない
活発さ
停滞
技術スタック
python

調査の方向性

リンクされたフォークの megatron/arguments.py の 352-356 行から始め、その検証を NVIDIA Megatron-LM pull request #288 と比較してください。エキスパート並列化が分散オプティマイザーとどのように相互作用するかを追跡し、その assertion が必要かどうかを文書化してください。できれば、再現した非互換性、または互換性の確認も含めてください。

索引モデルが issue の本文から書いたものです。

説明

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.

主要言語
Python
スター
1.6k
フォーク
228
PR マージ指標
30日以内にマージされた PR はありません

環境構築

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。