AmbigQA Qwen3.5-2B fine-tunes, LoRA
Collection
10 items • Updated
How to use myyycroft/Qwen3.5-2B-AmbigQA-lora-8-short-form-prompt-member-2 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3.5-2B")
model = PeftModel.from_pretrained(base_model, "myyycroft/Qwen3.5-2B-AmbigQA-lora-8-short-form-prompt-member-2")Ensemble member 2 (seed 2060) from run qwen35_2b_lora_8_small_lr_short_form_prompt.
Fine-tuned from Qwen/Qwen3.5-2B on AmbigQA.
Metrics below are computed on small fixed subsets, not full benchmarks.
Exact subset sizes: ambigqa=128, ifeval=64, mmlu=228.
Reported values are from the final training step (step=1620, epoch=3).
sewon/ambig_qa (config_name=light)6e667596df70f17ba3c8e7be4b7361f6be8b60f8train, validation=validationdev_fraction=0.1, split_seed=1729FalseYou answer open-domain factoid questions with a short-form answer only.
Reply with just the answer string: typically a name, place, organization, title, date, number, or brief factual phrase (often about 1–4 words). Do not write a sentence, explanation, or preamble.
Examples of the expected style:
- Who painted The Persistence of Memory? → Salvador Dalí
- When was the first iPhone released in the United States? → June 29, 2007
- What is the capital of New Zealand? → Wellington
- Who wrote Pride and Prejudice? → Jane Austen
- What is the chemical symbol for gold? → Au
- Who directed the 1994 film Pulp Fiction? → Quentin Tarantino
Formatting rules:
- Output the answer alone on a single line.
- Do not restate the question, add reasoning, markdown, bullets, or quotation marks around the answer.
- Prefer a canonical short form over a descriptive sentence.
- Give one answer only; do not list alternatives or hedge.
- Use natural short forms for dates and names (e.g. June 29, 2007; Jane Austen).
Exact resolved settings used for this run (also attached as resolved_config.yaml):
model:
name: Qwen/Qwen3.5-2B
revision: main
dtype: bfloat16
enable_thinking: false
dataset:
name: sewon/ambig_qa
config_name: light
revision: 6e667596df70f17ba3c8e7be4b7361f6be8b60f8
train_split: train
validation_split: validation
dev_fraction: 0.1
split_seed: 1729
bootstrap: false
max_train_examples: null
max_dev_examples: null
system_prompt_set: true
adaptation:
method: lora
lora:
r: 8
alpha: 16
dropout: 0.0
target_modules: all-linear
bias: none
training:
num_train_epochs: 3.0
max_steps: -1
learning_rate: 5.0e-05
weight_decay: 0.01
warmup_ratio: 0.03
per_device_train_batch_size: 16
per_device_eval_batch_size: 32
gradient_accumulation_steps: 2
max_seq_length: 512
logging_steps: 10
save_every_steps: 150
save_total_limit: 3
gradient_checkpointing: false
bf16: true
tf32: true
max_grad_norm: 1.0
dataloader_num_workers: 2
adam_beta1: 0.9
adam_beta2: 0.999
adam_epsilon: 1.0e-08
ensemble:
size: 5
base_seed: 42
this_member_index: 2
this_member_seed: 2060
evaluation_subsets:
ambigqa:
subset_size: 128
seed: 1001
ifeval:
subset_size: 64
seed: 1002
mmlu:
subset_size: 228
seed: 1004
| member | seed | steps | epochs | train_loss | AmbigQA acc | AmbigQA AlignScore | IFEval strict | MMLU acc |
|---|---|---|---|---|---|---|---|---|
| 0 | 42 | 1620 | 3 | 1.5179 | 0.0703 | 0.1238 | 0.5469 | 0.5833 |
| 1 | 1051 | 1620 | 3 | 1.5207 | 0.0781 | 0.1265 | 0.4531 | 0.6096 |
| 2 (this repo) | 2060 | 1620 | 3 | 1.5133 | 0.0703 | 0.1341 | 0.5781 | 0.6184 |
| 3 | 3069 | 1620 | 3 | 1.5196 | 0.0859 | 0.1321 | 0.5156 | 0.5877 |
| 4 | 4078 | 1620 | 3 | 1.5245 | 0.0703 | 0.1225 | 0.5312 | 0.5789 |
members/member_002/final/resolved_config.yaml — full resolved training configensemble_metrics.png — train/eval curves for the whole ensemble