Views
No views yet
| Parameter | Value |
|---|---|
| Learning Rate | 2e-5 |
| Weight Decay | 1e-4 |
| Epochs | 2 |
| Global Batch Size | 128 |
| Micro-batch Size per GPU | 1 |
| Gradient Accumulation Steps | 2 |
| Max Sequence Length | 32,768 tokens |
| Optimizer | AdamW (β₁=0.9, β₂=0.95) |
| LR Scheduler | Cosine with 10% warmup |
| Gradient Clipping | 1.0 |
| Attention | FlashAttention-2 |
| dtype | bfloat16 |
| Item | Value |
|---|---|
| GPU | H200 |
| Nodes | 8 |
| GPUs per Node | 8 |
| Total GPUs | 64 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained(
4 "ZhuofengLi/Qwen3.5-9B-Base-Nemotron-SFT-2220-steps",
5 torch_dtype="auto",
6 device_map="auto",
7)
8tokenizer = AutoTokenizer.from_pretrained("ZhuofengLi/Qwen3.5-9B-Base-Nemotron-SFT-2220-steps")