Skip to content

Commit 7225472

Browse files
suachongZixianWangAMDmmarcinkiewiczhXl3sShriyaRishab
authored
gpt-oss-20b gbs64 rcp logs (#869)
* add small llm pretraining address review from previous PR Credit co-authors for prior squash Co-authored-by: ZixianWangAMD <zixiwang@amd.com> Co-authored-by: Michal Marcinkiewicz <michalm@nvidia.com> Co-authored-by: Lukasz Pierscieniewski <l.pierscieniewski@gmail.com> * Update pretrain_llama31.py disable async save and save intermediate checkpoint * update README instruction, minor change to callback and set default target log perplexity to be 3.3 for consistency purposes * set LR for GB32 * update README with download instructions from https://training.mlcommons-storage.org/index.html * update link to https://github.com/mlcommons/r2-downloader * Revert "update link to https://github.com/mlcommons/r2-downloader" This reverts commit fed1bb4. * initial commit for gpt-oss-20b * update README * update NV dockerfile and readme * set random seed * update license for amd + clean up rocm dockerfile * revisit the target log perplexity after establishing rcp * remove target metric * remove HF_TOKEN * Update run_and_time.sh * Create run.sub * Update run_and_time.sh * Update README.md * Update run.sub * Update run.sub * Update run.sub * Update Dockerfile.nvidia * Update run.sub * update configs to match 8b, addressed pr comments * expose adam_eps with env var * update more configs based on hf + nvidia configs * remove yarn patch and add primus evaluator patch * update megatron validation consumed samples * remove utilities, not needed * set master port to 29500, consistent with src/train.py * remove hardcoded company name * Revise evaluation metrics and training parameters in README Updated README.md to clarify evaluation metrics and training parameters. * update target metric and approximate runtime * upload gbs16 logs * rm gpt-oss-20b folder * remove extra file * update approximate runtime * upload gbs32 rcp logs * upload gbs64 rcp logs --------- Co-authored-by: ZixianWangAMD <zixiwang@amd.com> Co-authored-by: Michal Marcinkiewicz <michalm@nvidia.com> Co-authored-by: Lukasz Pierscieniewski <l.pierscieniewski@gmail.com> Co-authored-by: Michał Marcinkiewicz <43240942+mmarcinkiewicz@users.noreply.github.com> Co-authored-by: Shriya Rishab <spalsamudram@nvidia.com>
1 parent 8346a11 commit 7225472

20 files changed

Lines changed: 98138 additions & 0 deletions

File tree

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_0.log

Lines changed: 4600 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_1.log

Lines changed: 4996 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_10.log

Lines changed: 5194 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_11.log

Lines changed: 4600 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_12.log

Lines changed: 4798 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_13.log

Lines changed: 5194 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_14.log

Lines changed: 4798 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_15.log

Lines changed: 4798 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_16.log

Lines changed: 5392 additions & 0 deletions
Large diffs are not rendered by default.

small_llm_moe_pretraining/primus/rcp_logs/gbs64/run_17.log

Lines changed: 4600 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)