Hugging Face's logo

unlearning-cleanslate
/
llama-3.1-8b-simnpo-baseline

Safetensors
llama
Model card Files
xet
 Community
llama-3.1-8b-simnpo-baseline
File size: 1,605 Bytes
bd18155
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
{
  "run_name": "simnpo_Llama-3.1-8B_20260423_063935",
  "run_dir": "/runs/llama-8b/baseline/simnpo/simnpo_Llama-3.1-8B_20260423_063935",
  "config": {
    "model_name": "meta-llama/Llama-3.1-8B",
    "forget_path": "unlearning-cleanslate/fsid_llama-8b:forget_baseline",
    "retain_path": "unlearning-cleanslate/fsid_llama-8b:retain_baseline",
    "output_dir": "/runs/llama-8b/baseline/simnpo",
    "objective": "simnpo",
    "num_epochs": 2,
    "batch_size": 2,
    "gradient_accum_steps": 16,
    "lr": 5e-05,
    "weight_decay": 0.0,
    "warmup_steps": 10,
    "max_grad_norm": 1.0,
    "forget_weight": 5.0,
    "retain_weight": 1.0,
    "retain_loss_type": "nll",
    "npo_beta": 2.0,
    "simnpo_delta": 0.25,
    "rmu_steering_coeff": 2.0,
    "rmu_module_regex": "model\\.layers\\.7",
    "rmu_control_vec_seed": 0,
    "undial_beta": 10.0,
    "validation_path": null,
    "eval_every": 50,
    "eval_max_samples": 200,
    "save_every": 0,
    "distributed_strategy": "fsdp",
    "fsdp_sharding_strategy": "full_shard",
    "fsdp_cpu_offload": false,
    "device": "auto",
    "dtype": "bfloat16",
    "seed": 42,
    "num_workers": 0,
    "log_every": 10,
    "gradient_checkpointing": false,
    "torch_compile": false
  },
  "num_forget_rows": 2407,
  "num_forget_pairs": 2407,
  "num_retain_rows": 1646,
  "num_retain_sequences": 1646,
  "num_val_sequences": 0,
  "num_trainable_params": 2007565312,
  "num_total_params": 2007565312,
  "run_seconds": 183.5273723602295,
  "checkpoint_dir": "/runs/llama-8b/baseline/simnpo/simnpo_Llama-3.1-8B_20260423_063935",
  "is_main_process": true
}