README: large model
Browse files
README.md
CHANGED
|
@@ -9,6 +9,7 @@ Files downloaded by the Android app (github.com/cyttic/android-hwrg), which runs
|
|
| 9 |
| File | What |
|
| 10 |
|---|---|
|
| 11 |
| `line_int8.onnx` | CTC line model `cyttic/exp35-trocr-base-en-ctc-seqhead-allreal-synpunct` (TrOCR-base ViT encoder + 2-layer transformer CTC head, 198 classes), ONNX opset 17, dynamic int8 (all constant-weight MatMuls except the ViT layer-2 MLP fc2, kept in fp32). Inputs: `pixel_values` (B, 3, 384, 384) in [-1, 1], `cols` (B,) int64; output `log_probs` (B, 288, 198). |
|
|
|
|
| 12 |
| `charset.json` | `chars[k]` = character of class k; `chars[0]` = CTC blank. |
|
| 13 |
| `heb_char6_kn.bin.gz` | Character 6-gram LM, interpolated Kneser-Ney (Ben-Yehuda + DictaGen text), binary layout in `tools/export_lm.py`. |
|
| 14 |
|
|
|
|
| 9 |
| File | What |
|
| 10 |
|---|---|
|
| 11 |
| `line_int8.onnx` | CTC line model `cyttic/exp35-trocr-base-en-ctc-seqhead-allreal-synpunct` (TrOCR-base ViT encoder + 2-layer transformer CTC head, 198 classes), ONNX opset 17, dynamic int8 (all constant-weight MatMuls except the ViT layer-2 MLP fc2, kept in fp32). Inputs: `pixel_values` (B, 3, 384, 384) in [-1, 1], `cols` (B,) int64; output `log_probs` (B, 288, 198). |
|
| 12 |
+
| `line_large_int8.onnx` | CTC line model `cyttic/exp35-trocr-large-en-ctc-seqhead-allreal-synpunct` (TrOCR-large ViT encoder, 24 layers + 2-layer transformer CTC head), same inputs/outputs; dynamic int8 except the ViT layer-0 MLP fc2 (fp32). Decoded with the punctuation-aware LM fusion (alpha 0.3, beta 0.6, LM on Hebrew letters + space). |
|
| 13 |
| `charset.json` | `chars[k]` = character of class k; `chars[0]` = CTC blank. |
|
| 14 |
| `heb_char6_kn.bin.gz` | Character 6-gram LM, interpolated Kneser-Ney (Ben-Yehuda + DictaGen text), binary layout in `tools/export_lm.py`. |
|
| 15 |
|