48 lines
1.5 KiB
JSON
48 lines
1.5 KiB
JSON
{
|
|
"run_name": "rmu_Llama-3.1-8B_20260724_012047",
|
|
"run_dir": "results/rebuttal/train/llama-3_1-8b-rmu-bm25-6t/rmu_Llama-3.1-8B_20260724_012047",
|
|
"config": {
|
|
"model_name": "meta-llama/Llama-3.1-8B",
|
|
"forget_path": "unlearning-cleanslate/fsid-curated-llama-8b:forget:bm25_6T",
|
|
"retain_path": "unlearning-cleanslate/fsid-curated-llama-8b:retain:bm25_6T",
|
|
"output_dir": "results/rebuttal/train/llama-3_1-8b-rmu-bm25-6t",
|
|
"objective": "rmu",
|
|
"num_epochs": 2,
|
|
"batch_size": 8,
|
|
"gradient_accum_steps": 4,
|
|
"lr": 1e-05,
|
|
"weight_decay": 0.0,
|
|
"warmup_steps": 20,
|
|
"max_grad_norm": 1.0,
|
|
"forget_weight": 1.0,
|
|
"retain_weight": 1.0,
|
|
"retain_loss_type": "embed_diff",
|
|
"npo_beta": 0.1,
|
|
"simnpo_delta": 0.0,
|
|
"rmu_steering_coeff": 2.0,
|
|
"rmu_module_regex": "model\\.layers\\.7",
|
|
"rmu_control_vec_seed": 0,
|
|
"undial_beta": 10.0,
|
|
"save_every": 0,
|
|
"fsdp_sharding_strategy": "full_shard",
|
|
"dtype": "bfloat16",
|
|
"seed": 42,
|
|
"num_workers": 0,
|
|
"log_every": 10,
|
|
"gradient_checkpointing": false,
|
|
"torch_compile": false,
|
|
"push_to_hub": false,
|
|
"hub_repo_id": null,
|
|
"hub_private": false,
|
|
"optim": "adamw_torch"
|
|
},
|
|
"num_forget_rows": 2901,
|
|
"num_forget_pairs": 2901,
|
|
"num_retain_rows": 1646,
|
|
"num_retain_sequences": 1646,
|
|
"num_trainable_params": 8030261248,
|
|
"num_total_params": 8030261248,
|
|
"run_seconds": 170.28291535377502,
|
|
"checkpoint_dir": "results/rebuttal/train/llama-3_1-8b-rmu-bm25-6t/rmu_Llama-3.1-8B_20260724_012047",
|
|
"is_main_process": true
|
|
} |