bigcode

/

starcoder2-15b

like 568

Follow

BigCode 937

I noticed that StarCoder2 uses two stages of pre-training, with the stage 2 for long-context training.

Take StarCoder-15B as an example,

• In the stage 1, rope_theta=1e4, warmup=1000, max_lr=3e-4
• In the stage 2, rope_theta=1e5, max_lr=3e-5

So the question is:

• In the stage 1, what is min_lr?
• In the stage 2, what is min_lr and warmup?