初始化项目,由ModelHub XC社区提供模型
Model: anime-sh/qwen3-8b-rmu-bm25-10b-rebuttal Source: Original Platform
This commit is contained in:
48
train_summary.json
Normal file
48
train_summary.json
Normal file
@@ -0,0 +1,48 @@
|
||||
{
|
||||
"run_name": "rmu_Qwen3-8B_20260724_021302",
|
||||
"run_dir": "results/rebuttal/train/qwen3-8b-rmu-bm25-10b/rmu_Qwen3-8B_20260724_021302",
|
||||
"config": {
|
||||
"model_name": "Qwen/Qwen3-8B",
|
||||
"forget_path": "unlearning-cleanslate/fsid-curated-qwen3-8b:forget:bm25_10B",
|
||||
"retain_path": "unlearning-cleanslate/fsid-curated-qwen3-8b:retain:bm25_10B",
|
||||
"output_dir": "results/rebuttal/train/qwen3-8b-rmu-bm25-10b",
|
||||
"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": 6326,
|
||||
"num_forget_pairs": 6326,
|
||||
"num_retain_rows": 1646,
|
||||
"num_retain_sequences": 1646,
|
||||
"num_trainable_params": 8190735360,
|
||||
"num_total_params": 8190735360,
|
||||
"run_seconds": 426.0725100040436,
|
||||
"checkpoint_dir": "results/rebuttal/train/qwen3-8b-rmu-bm25-10b/rmu_Qwen3-8B_20260724_021302",
|
||||
"is_main_process": true
|
||||
}
|
||||
Reference in New Issue
Block a user