This model is a fine-tuned version of facebook/wav2vec2-base-960h for Speech Emotion Recognition (SER). It has been trained using a Frozen Feature Extractor strategy to preserve the model's acoustic understanding while adapting to emotion detection. This approach ensures stable performance and prevents "Catastrophic Forgetting," achieving nearly 80% accuracy on the validation set. Update: The "Calm" and "Neutral" classes have been merged to improve classification consistency, resulting in 7 distinct emotion classes. The model was trained on a combined dataset of ~12,000 audio files from: The model classifies audio into one of the following emotions: 1. Angry 2. Disgust 3. Fear 4. Happy 5.…
keyword is an open-weight model for audio classification from Yim. It has 95M parameters. At 16-bit it needs about 0.2 GB of GPU memory, which fits on 1x MI300X from $1.85 an hour, at the lowest prices in the SAVRN Index.
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset.
Runs On
What it takes to serve keyword (95M parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.
| Precision | Weights | Memory needed | Cheapest setup | Per hour | Also fits |
|---|---|---|---|---|---|
| 16-bit | 0.2 GB | 0.2 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
| 8-bit | 0.1 GB | 0.1 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
| 4-bit | 0.0 GB | 0.1 GB | 1x MI300X (192 GB) Vultr |
$1.85 | 1x H100 $1.99 · 1x MI325X $2.00 |
Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Oct 18, 2026.
keyword on every accelerator the SAVRN Index prices, at every precision
Model Card
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 4 - evalbatchsize: 4 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cpu - Datasets 4.8.5 - Tokenizers 0.23.2
Excerpt from the card by Yim.
Configuration
- Architecture
- Wav2Vec2ForSequenceClassification
- Layers
- 12
- Hidden size
- 768
- Feed-forward size
- 3,072
- Attention heads
- 12
- Vocabulary size
- 56
- Model type
- wav2vec2
Identity and Version
- Repository
- bbangju/keyword
- Publisher
- Yim
- Task
- Audio classification
- Modality
- Audio
- Library
- transformers
- Parameters
- 95M parameters
- Languages
- Not stated by the source
- Revision
- b6924cd379607108e43a5089e534e40892967d16
- First published
- 2026-10-13
- Last updated
- 2026-10-13
Files and Weights
6 files, 378.3 MB in total. The weights are 2 files totalling 378.3 MB in bin, safetensors.
Every file
| File | Type | Size | SHA-256 |
|---|---|---|---|
| model.safetensors | Weights | 378.3 MB | b062f2aa6d46 |
| training_args.bin | Weights | 5.2 KB | 198d8c4848f3 |
| config.json | Configuration | 3.1 KB | — |
| preprocessor_config.json | Configuration | 212 B | — |
| README.md | Documentation | 2.1 KB | — |
| .gitattributes | Repository | 1.5 KB | — |
License and Download
- License
- Not stated by the source
- Access
- Open weights, no gate
- Download size
- 378.3 MB
Released by Yim through its official repository on Hugging Face.
Built From
- Derived from Kkonjeong/wav2vec2-base-korean
Memory Requirements
| Precision | Weights in memory |
|---|---|
| As published | 378.3 MB |
| 16-bit | 0.2 GB |
| 8-bit | 0.1 GB |
| 4-bit | 0.0 GB |
Weights only, from the published parameter count; the key-value cache and runtime add to this.
Questions About keyword
How much GPU memory does keyword need?
About 0.2 GB at 16-bit and 0.1 GB at 4-bit: the weights (95M parameters) plus a working margin. A long context needs more.
What is the cheapest GPU to run keyword on?
At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.
Similar Models
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 4 - evalbatchsize: 4 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cpu - Datasets 4.8.5 - Tokenizers 0.23.2
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 8 - evalbatchsize: 8 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cu130 - Datasets 4.8.5 - Tokenizers 0.23.2
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 8 - evalbatchsize: 8 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cu130 - Datasets 4.8.5 - Tokenizers 0.23.2
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 8 - evalbatchsize: 8 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cpu - Datasets 4.8.5 - Tokenizers 0.23.2
This model is a fine-tuned version of Kkonjeong/wav2vec2-base-korean on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 0.0001 - trainbatchsize: 8 - evalbatchsize: 8 - lrschedulertype: linear - numepochs: 10 - mixedprecisiontraining: Native AMP - Transformers 5.18.0 - Pytorch 2.11.0+cpu - Datasets 4.8.5 - Tokenizers 0.23.2