doctr-fast-base / validation.json
tomsanbear's picture
Add validated docTR FAST base bundle
6766e21 verified
Raw
History Blame Contribute Delete
2.51 kB
{
"conversion_tool": "tools/models/candle/convert_fast.py@schema-v1",
"dtype_device_results": [
{
"device": "cpu",
"dtype": "f32",
"note": "checkpoint conversion and shape contract smoke; native Candle forward parity lands in impl-candle-ocr-detectors",
"status": "passed"
},
{
"device": "cuda",
"dtype": "f16",
"note": "accelerator validation deferred until native FAST forward exists",
"status": "unvalidated"
},
{
"device": "metal",
"dtype": "f16",
"note": "accelerator validation deferred until native FAST forward exists",
"status": "unvalidated"
}
],
"generated_files": [
{
"path": "model.safetensors",
"sha256": "60779443cece1dee99513f27486343cf41b02c63a5a989e9ffac961119dd791d",
"size_bytes": 65610021
},
{
"path": "config.json",
"sha256": "de4adca58469d0a217e3c53f18c6e813072e62cebf4912408fbcf991008f4465",
"size_bytes": 5786
},
{
"path": "preprocessor_config.json",
"sha256": "360a486f52484e2a256e6d37b22ea58ca91b695c0f2003099ff039175260f701",
"size_bytes": 527
},
{
"path": "tensor_map.json",
"sha256": "3dece6ea0c4caabb12049a6ffc5b0f0631e536980d8989bb64642ed87dd10237",
"size_bytes": 165363
},
{
"path": "bundle.json",
"sha256": "d68958b9b6f32ab5088c111be0eaeeef2a9b88eea0c96ad9b19a719537b9f0f2",
"size_bytes": 1744
}
],
"parity_fixture_ids": [
"onnxtr-fast-base-synthetic-gray-logits",
"onnxtr-fast-base-marker-english-dense-prob-map"
],
"schema_version": 1,
"source_files": [
{
"path": "fast_base-688a8b34.pt",
"sha256": "688a8b3489e9f5d0290c476c6272ec3b18de3ee646c8a0dc158203b1a9c62ace",
"size_bytes": 65814772
},
{
"path": "doctr/models/detection/fast/pytorch.py"
},
{
"path": "doctr/models/classification/textnet/pytorch.py"
},
{
"path": "doctr/models/modules/layers/pytorch.py"
}
],
"tested_io": [
{
"id": "doctr-fast-base-cpu-f32-shape-smoke",
"inputs": [
{
"dtype": "f32",
"name": "input",
"shape": [
1,
3,
1024,
1024
]
}
],
"outputs": [
{
"dtype": "f32",
"name": "logits",
"shape": [
1,
1,
1024,
1024
]
}
]
}
]
}