Hacktoberfest 2026:メンテナが10月に向けて印を付けた、オープンで初心者向けの issue。 Hacktoberfest の issue を見る

different load_balancing_loss with different pipeline_parallel_size

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

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

評価

難易度
4/5
見積もり時間
3〜5日
初心者へのやさしさ
35/100
issue の種類
バグ
明瞭さ
おおむね明確
活発さ
停滞
技術スタック
python

調査の方向性

報告された64エキスパートの500M GPT構成を再現し、pipeline_parallel_sizeを1、2、4の間で変化させたうえで、load_balancing_lossを報告された値と比較します。差異が数値精度によるものか、パイプライン並列の集約に関する問題によるものかを調査します。異なる値を説明し、バグが存在するかどうかを確認できれば完了です。

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

説明

question

I load the same model trained with megatron + megablocks, and I found the load_balancing_loss is slightly different. When I increase the pipeline_parallel_size, the load_balancing_loss is also increasing. Is it just the problem of precision or there is a potential bug?

For example, when I train a 500M gpt model with 64 experts, I list the lbl and pp_size in the table below.

pp_size lbl
1 1.005E-01
2 1.007E-01
4 1.013E-01
主要言語
Python
スター
1.6k
フォーク
230
PR マージ指標
30日以内にマージされた PR はありません

環境構築

はじめの一歩

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

databricks/megablocks のほかの issue

databricks/megablocks の issue をすべて見る

似ている issue

Python の issue をもっと見る

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

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