ONNX Model Zoo

Vorrapard/wav2vec2-command-classifier

By Vorrapard · ↓ 0 · ♥ 0 · Updated 2025-05-14

What's in this repo

This page lists the files in Vorrapard/wav2vec2-command-classifier, including wav2vec2_command_classifier.onnx.

  • 1 ONNX models (.onnx files)
  • 5 files, including 1 .md, 1 .json, 1 .bin, 1 .onnx
  • Last updated 2025-05-14

ONNX files

File Size Download
wav2vec2_command_classifier.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

Command Classifier

This is a lightweight classifier model trained on top of Wav2Vec2 features for classifying speech commands.

Usage

You can use this classifier by combining it with Wav2Vec2 features. The classifier expects mean-pooled Wav2Vec2 hidden states.

from transformers import Wav2Vec2Model
from command_classifier import CommandClassifier
import torch

wav2vec = Wav2Vec2Model.from_pretrained("facebook/wav2vec2-base")
classifier = CommandClassifier(num_classes=35)
classifier.load_state_dict(torch.load("pytorch_model.bin"))

Model info

pytorch onnx custom dataset:google/speech_commands arxiv:1804.03209 base_model:facebook/wav2vec2-base base_model:quantized:facebook/wav2vec2-base license:mit region:us
↓ 0 ♥ 0

Created: 2025-04-23

Updated: 2025-05-14

View on Hugging Face

Files (5)

.gitattributes
README.md
config.json
pytorch_model.bin
wav2vec2_command_classifier.onnx ONNX

Related models

More from Vorrapard

No other models by this author.

More models

No task type declared.

Popular models