msluszniak's picture
Add spec-compliant config.json files
0914368 verified
Raw
History Blame Contribute Delete
2.64 kB
{
"$schema": "https://huggingface.co/software-mansion/react-native-executorch-spec/resolve/main/config.schema.json",
"model": "yolo26_pose",
"family": "yolo",
"capabilities": [
"object-detection",
"instance-segmentation"
],
"backend": "xnnpack",
"license": "agpl-3.0",
"variants": [
{
"file": "yolo26_pose_n_xnnpack_fp32.pte",
"precision": "fp32",
"quantized": false,
"default": true,
"methods": {
"forward_384": {
"inputs": [
{
"shape": [
1,
3,
384,
384
],
"dtype": "float32"
}
],
"outputs": [
{
"shape": [
300,
4
],
"dtype": "float32"
},
{
"shape": [
300
],
"dtype": "float32"
},
{
"shape": [
300,
17,
3
],
"dtype": "float32"
}
]
},
"forward_512": {
"inputs": [
{
"shape": [
1,
3,
512,
512
],
"dtype": "float32"
}
],
"outputs": [
{
"shape": [
300,
4
],
"dtype": "float32"
},
{
"shape": [
300
],
"dtype": "float32"
},
{
"shape": [
300,
17,
3
],
"dtype": "float32"
}
]
},
"forward_640": {
"inputs": [
{
"shape": [
1,
3,
640,
640
],
"dtype": "float32"
}
],
"outputs": [
{
"shape": [
300,
4
],
"dtype": "float32"
},
{
"shape": [
300
],
"dtype": "float32"
},
{
"shape": [
300,
17,
3
],
"dtype": "float32"
}
]
}
}
}
],
"size": "n"
}