File size: 2,582 Bytes
8622b7d | 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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 | {
"source_checkpoint": "runs/best_step402",
"base_model": "Qwen/Qwen3.5-9B",
"dtype": "bfloat16",
"format": "markers-v3-plainchains",
"head_dim": 256,
"temperature": 1.8589280843734741,
"stage": "cispo",
"step": 402,
"sources": {
"banking77": 0,
"boolq": 1,
"agnews": 2,
"mnli": 3,
"sst5": 4,
"yelp": 5,
"trec": 6,
"dbpedia14": 7,
"amazon": 8,
"imdb": 9,
"mmlu": 10,
"emotion": 11,
"tweet_offensive": 12,
"qnli": 13,
"paws": 14,
"sciq": 15,
"contrastive": 16,
"compositional": 17,
"composition_holdout": 18,
"semif": 19,
"wanli": 20,
"mmlu_pro": 21,
"buried": 22,
"unknowable": 23,
"unknowable_control": 24,
"night2_assertion": 25,
"night2_dates": 26,
"night2_unknowable": 27,
"night2_unknowable_control": 28
},
"metrics": null,
"training_config": {
"stage": "cispo",
"run_dir": "runs/cispo_rep",
"data_dir": "data",
"model": "Qwen/Qwen3.5-9B",
"init": "runs/sft_v4/final",
"resume": null,
"lora_r": 16,
"lora_alpha": 32.0,
"lora_dropout": 0.0,
"head_dim": 256,
"lr": 5e-05,
"head_lr": 0.0001,
"weight_decay": 0.0,
"warmup": 20,
"epochs": 4,
"max_steps": 0,
"batch_size": 8,
"grad_clip": 1.0,
"max_len": 4096,
"max_seq_len": 8192,
"train_limit": 0,
"dev_limit": 0,
"eval_every": 50,
"eval_batch_size": 16,
"eval_think": true,
"eval_think_limit": 256,
"log_every": 10,
"seed": 0,
"gradient_checkpointing": true,
"prompts_per_step": 8,
"group": 8,
"nothink_per_group": 0,
"nothink_start_epoch": 1,
"max_think": 2560,
"temperature": 1.0,
"top_p": 1.0,
"eps_high": 0.2,
"updates": 1,
"train_mix_frac": 0.0,
"rollout_ce_weight": 0.5,
"anchor_ce_weight": 0.5,
"anchor_batch": 8,
"gen_batch_size": 64,
"logprob_chunk": 128,
"micro_groups": 2,
"length_hinge": 2048,
"length_hinge_end": 1024,
"length_ramp": 0,
"length_penalty_weight": 1.0,
"length_penalty_cap": 0.1,
"length_penalty_correct_only": true,
"length_hinge2": 0,
"length_hinge2_end": 1792,
"length_penalty2_weight": 0.0,
"closure_reward": 0.0,
"pad_multiple": 128,
"num_workers": 2,
"compile": true,
"graph_decode": true,
"async_rollouts": true,
"ckpt_minutes": 45.0,
"log_rollouts": 3,
"log_chain_chars": 1200,
"sft_think_frac": 0.5,
"chain_cache": "data/chains",
"filter_active": false,
"filter_min_spread": 0.02
}
}
|