haotiansun014 commited on
Commit
3d81379
·
verified ·
1 Parent(s): 8e9133f

Add README

Browse files
Files changed (1) hide show
  1. README.md +52 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language: en
4
+ tags:
5
+ - draft-refine
6
+ - block-diffusion
7
+ - nce
8
+ - qwen3-4b
9
+ ---
10
+
11
+ # Qwen3-4B Stage-3 NCE — temp07-perround variant
12
+
13
+ Stage-3 Noise-Contrastive-Estimation training resumed from the Stage-2
14
+ end ckpt at step 7335. NCE phase trains the scorer head to rank K=4
15
+ candidate completions per block. Differs from beam-perround in proposal
16
+ sampling: this variant uses `softmax_sampling` at proposal_temperature
17
+ =0.7 (vs argmax in beam-perround) — closer to training-time scorer
18
+ noise.
19
+
20
+ ## Files
21
+
22
+ | File | Size |
23
+ |---|---|
24
+ | `model.pt` | 8.5 GB |
25
+ | `optimizer.pt` | 0.45 GB |
26
+ | `scheduler.pt` | 1.7 KB |
27
+ | `eval_batches.pt` | 13 MB |
28
+ | `rng_rank{0..23}.pt` | 14.7 KB each |
29
+
30
+ Total: ~9.54 GB / 30 files. Full resume state for re-training.
31
+
32
+ ## Step + lineage
33
+
34
+ - Resume from: Stage-2 ckpt at step 7335
35
+ - This ckpt: step 9124 (1789 NCE-phase steps)
36
+ - Proposal: `combine=softmax_sampling`, `proposal_temperature=0.7`
37
+ - Config: `configs/large_scale/qwen3_4b_stage3_nce_resume7335_temp07_perround_6n.yaml`
38
+
39
+ ## Note on inference reproducibility
40
+
41
+ When evaluating this ckpt, use:
42
+ ```
43
+ combine=softmax_sampling proposal_temperature=0.7
44
+ ```
45
+ to MATCH the training distribution. Using `combine=beam_bayes argmax`
46
+ (as for the perround/lastround variants) on this ckpt produces an
47
+ out-of-distribution scorer signal.
48
+
49
+ ## Related archives
50
+
51
+ - `haotiansun014/qwen3-4b-stage3-nce-7335-perround-archive`
52
+ - `haotiansun014/qwen3-4b-stage3-nce-7335-lastround-archive`