AmbigQA Qwen3.5-2B fine-tunes, OFT
Collection
10 items • Updated
How to use myyycroft/Qwen3.5-2B-AmbigQA-oft-block-32-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-oft-block-32-member-2")Ensemble member 2 (seed 2060) from run qwen35_2b_oft_block_32_small_lr.
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=1729FalseNo system prompt (dataset.system_prompt: null).
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: false
adaptation:
method: oft
oft:
target_modules: all-linear
r: null
block_size: 32
module_dropout: 0.05
use_cayley_neumann: true
num_cayley_neumann_terms: 5
coft: false
eps: 6.0e-05
block_share: false
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.5178 | 0.0703 | 0.1287 | 0.5938 | 0.6228 |
| 1 | 1051 | 1620 | 3 | 1.5202 | 0.0703 | 0.1171 | 0.5312 | 0.6228 |
| 2 (this repo) | 2060 | 1620 | 3 | 1.5141 | 0.0625 | 0.1168 | 0.5938 | 0.6228 |
| 3 | 3069 | 1620 | 3 | 1.5190 | 0.0703 | 0.1151 | 0.5625 | 0.6447 |
| 4 | 4078 | 1620 | 3 | 1.5222 | 0.0547 | 0.1138 | 0.5312 | 0.6272 |
members/member_002/final/resolved_config.yaml — full resolved training configensemble_metrics.png — train/eval curves for the whole ensemble
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-oft-block-32-member-2")