Views
No views yet
| Item | Value |
|---|---|
| Base model | Qwen/Qwen3.5-0.8B |
| Task | Reddit post → short TL;DR-style summary |
| Training type | Causal LM SFT (completion-only cross-entropy) |
| Selected checkpoint | End of epoch 1 (global step 1292) |
Validation eval_loss (epoch 1) | ~1.868 |
CarperAI/openai_summarize_tldr (train / valid)AL_Uncertainty/src/data_utils.py):
rstrip)r/offmychest or r/tifulabel (reference TL;DR) as the assistant turn in a chat template.sft/config/default.yaml):| Hyperparameter | Value |
|---|---|
| Epochs (planned) | 3 |
| Checkpoint released | Epoch 1 only |
| Per-device train batch size | 2 |
| Gradient accumulation | 8 → effective batch 16 |
| Learning rate | 2e-5 |
| Warmup ratio | 0.05 |
| Max sequence length | 2048 |
| Precision | bf16 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "Maximus200005/qwen35-0.8b-tldr-sft-ep1"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 trust_remote_code=True,
10 torch_dtype=torch.bfloat16,
11 device_map="auto",
12 use_safetensors=True,
13)