You can deploy this model and start earning money today!

Jeesup/
llama32-1B-mnli-bf16-lora-seed44

Name: llama32-1B-mnli-bf16-lora-seed44
Availability: InStock

meta-llama/Llama-3.2-1B

adapter

Views

No views yet

llama32-1B-mnli-bf16-lora-seed44

LoRA adapter trained on GLUE MNLI on top of a bf16 backbone of meta-llama/Llama-3.2-1B.

Part of a controlled study of whether the backbone bit-width changes what a LoRA adapter learns. For a given (model size, seed) the adapter initialisation is identical across the bf16 / int8 / nf4 arms, and the data order, optimiser, schedule and LoRA hyperparameters are held fixed — so any difference in the learned update is attributable to the backbone.

Result

metric	validation	test
accuracy	0.8640	0.8660
macro-F1	0.8637	0.8657
loss	0.4084	0.4223

Test-set majority-class baseline: 0.3630

peak GPU memory: 3.90 GiB
training time: 16.0 min (936 steps)
GPU: NVIDIA GeForce RTX 4090

Setup

seed: 44 · adapter init: shared:lora_init_1B_seed44.pt:128tensors
LoRA: r=16, alpha=32, dropout=0.0, bias=none, target_modules=['q_proj', 'k_proj', 'v_proj', 'o_proj']
trainable params: 3,407,872
epochs 3, lr 0.0002, max_len 256, batch 4 x grad_accum 16, cosine schedule, warmup 0.03

Prompt format

Trained as causal LM with the loss on the answer letter only (prompt tokens masked to -100):

Premise: ...
Hypothesis: ...

Does the premise entail the hypothesis?

A. Entailment
B. Neutral
C. Contradiction

Answer:

Evaluated by conditional likelihood over the answer letters (Entailment, Neutral, Contradiction).

GLUE test is unlabeled, so the official validation split is used as TEST and the validation set is carved from train (disjoint).

Jeesup/llama32-1B-mnli-bf16-lora-seed44

llama32-1B-mnli-bf16-lora-seed44

Result

Setup

Prompt format

Jeesup/
llama32-1B-mnli-bf16-lora-seed44