bigscience-workshop / bigscience-workshop/Megatron-DeepSpeed

How to continue pre-training Bloom?

Open
#366 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.4k
Forks
226
PR merge metrics
No merged PRs in 30d

Description

Hi
I'm trying to continue pre-training the bloom-560m on my own dataset on a single GPU.
I modified [this script](https://github.com/bigscience-workshop/Megatron-DeepSpeed/blob/main/examples/pretrain_gpt_single_node.sh) to fit my case.
However, i cannot figure out how to load the [checkpoint](https://huggingface.co/bigscience/bloom-560m-optimizer-states).

Is there any guide for what i'm doing?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.