Koshur-OCR / config.json
KashlabsAI's picture
Release Koshur OCR v2.1 — prefix beam + train-only trigram LM
e186a72 verified
Raw
History Blame Contribute Delete
2.11 kB
{
"release": "v2.1.0",
"weights_release": "v2.0.0",
"run_name": "ctc-100k-control-p0-20260729T061125Z-retry1",
"training_wandb_run": "https://wandb.ai/kash-ai-labs-kash-labs/koshur-pixel-ocr/runs/ctc-100k-control-p0-20260729T061125Z-retry1",
"beam_eval_wandb_run": "https://wandb.ai/kash-ai-labs-kash-labs/koshur-pixel-ocr/runs/ctc-v2-prefix-beam-20260729T152530Z",
"train_rows": 100000,
"val_rows": 1000,
"val_dedup_rows": 920,
"trained_epochs": 38,
"best_epoch": 18,
"batch_size": 32,
"learning_rate": 0.0007,
"weight_decay": 0.0001,
"temporal_dropout_p": 0.0,
"seed": 11,
"image_height": 64,
"image_width": 2048,
"canvas_align": "left",
"horizontal_stride": 4,
"cnn_channels": [48, 96, 160, 192],
"rnn_hidden": 256,
"rnn_layers": 2,
"rnn_dropout": 0.05,
"character_classes": 154,
"ctc_classes_including_blank": 155,
"parameters": 4475931,
"default_decoder": "ctc_prefix_beam",
"beam_decoder": {
"beam_width": 10,
"token_topk": 12,
"lm_type": "character_ngram",
"lm_order": 3,
"lm_smoothing": 0.1,
"lm_training_rows": 100000,
"lm_uses_validation_text": false,
"lm_weight": 0.4,
"token_bonus": 1.5,
"calibration_rows": 200,
"heldout_rows": 720
},
"snapshot_relative_path": "kp_scale/sentence_line-r874c4df69c3f-n100000-1000-len90",
"snapshot_images_mirrored": true,
"images_unmirrored_in_memory": true,
"labels_reversed_for_ltr_ctc": true,
"dataset_images_uploaded": false,
"selection_metric": "val_dedup/cer",
"metrics": {
"greedy_val_dedup_cer": 0.22617465745964668,
"greedy_val_dedup_exact_rate": 0.2554347826086957,
"beam_val_dedup_cer": 0.2135404186725336,
"beam_val_dedup_exact_rate": 0.2619565217391304,
"beam_heldout_cer": 0.21242690463653244,
"greedy_heldout_cer": 0.2243715877282931,
"beam_insertions": 736,
"beam_deletions": 4615,
"beam_substitutions": 4452
},
"checkpoint_sha256": "38382437ace33611c15ddbbf718ddcd88bf25f31092883bdcab540734199648d",
"language_model_sha256": "fca57843173a98321054cef4afeeac88a83f12a37191ee548103d909ec5564e3"
}