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.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support