CoolFace
Modelpublic

spadeMIA/pythia-1.4b-goodwiki-lora-r64

sourceHugging Faceupdated 1mo agoView on Hugging Face
0likes354downloads
Model Card

adapter

LoRA-fine-tuned checkpoint produced for membership inference research at the SPADE Lab, Koç University.

Base model

  • —Model: EleutherAI/pythia-1.4b
  • —PEFT method: LoRA

Training data

  • —Dataset: `spadeMIA/GoodWiki_Corpus_1024_2040`
  • —Split used: train (10,000 rows, all membership-positive)
  • —Stored text token range: 1024-2039 tokens (Pythia tokenizer family), plus one appended EOS token
  • —Max training sequence length: 2048 tokens
  • —Observed token stats: min=1024, mean=1840.7, median=1926.0, max=2039

LoRA configuration

HyperparameterValue
Rank (r)64
Alpha128
Dropout0.05
Target modulesquerykeyvalue, dense, densehto4h, dense4htoh

Training procedure

HyperparameterValue
Epochs2
Learning rate0.0002 (constant)
LR schedulerconstantwithwarmup
Warmup ratio0.01
Weight decay0.0
Max grad norm1.0
Per-device batch size1
Gradient accumulation steps32
Effective batch size32
Precisionbf16
TF32 matmulTrue
Gradient checkpointingTrue
Optimizeradamw_torch
Seed42
Checkpointingnone during training; only the final adapter is saved

Notes

  • —The corresponding held-out test split was not loaded or trained on.
  • —Eval-loss logging (monitoring only, no checkpoint selection) used train[:50].

Generated automatically by `run_lora.py` on 2026-08-10.