efficientnet_b0 for VKNN โ fp16, full-GPU
EfficientNet-B0 is the base of the compound-scaled EfficientNet family (~5.3M parameters), strong accuracy for its size. 224x224 RGB in, 1000 ImageNet class scores out.
EfficientNet-B0 image classifier (torchvision export), compiled for the VKNN Vulkan inference engine. Every op runs on the GPU (fp16 weights + fp16 compute, fp32 host boundary); the CPU backend serves as the bit-accuracy oracle in VKNN's device gates, and this model's GPU outputs are gated against an fp32 onnxruntime golden (cosine > 0.999, argmax agreement) on-device.
| File | efficientnet_b0_fp16.vxm (10M) |
| Input | input [1x3x224x224] (NCHW fp32) |
| Container | VXM3 (single graph, weights embedded) |
Run
# any VKNN example binary, e.g. the IO runner:
./vknn_run_io efficientnet_b0_fp16.vxm out_dir input.bin
Recompile from your own ONNX with vknn_compile model.onnx out.vxm --fp16.