Views
No views yet
\boxed{}.Please reason step by step, and put your final answer within \boxed{}.1messages = [
2 {"role": "system", "content": "Please reason step by step, and put your final answer within \\boxed{}."},
3 {"role": "user", "content": "Solve the problem here."},
4]| Item | Value |
|---|---|
| Base model | deepseek-ai/DeepSeek-R1-Distill-Qwen-14B |
| Dataset | RabotniKuma/Fast-Math-R1-SFT |
| Training type | Full-parameter SFT |
| GPUs used | 6 x NVIDIA H200 |
| Per-device batch size | 1 |
| Gradient accumulation | 8 |
| Effective global batch size | 48 |
| Epochs | 10 |
| Max sequence length | 24,000 tokens |
| Packing | Enabled |
| Learning rate | 1e-5 |
| Scheduler | Cosine |
| Precision | bfloat16 |
| Distributed setup | DeepSpeed ZeRO-3 |
analokmaus/kaggle-aimo2-fast-math-r1, adapted for this model and dataset.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "zbeeb/deepseek-r1-distill-qwen-14b-fast-math-r1-sft-10ep"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype="auto",
9 device_map="auto",
10 trust_remote_code=True,
11)
12
13messages = [
14 {"role": "system", "content": "Please reason step by step, and put your final answer within \\boxed{}."},
15 {"role": "user", "content": "What is 17 * 23?"},
16]
17
18inputs = tokenizer.apply_chat_template(
19 messages,
20 add_generation_prompt=True,
21 return_tensors="pt",
22).to(model.device)
23
24outputs = model.generate(
25 inputs,
26 max_new_tokens=2048,
27 temperature=0.6,
28 top_p=0.95,
29 do_sample=True,
30)
31
32print(tokenizer.decode(outputs[0], skip_special_tokens=True))