GGUF quantizations of the CED family (Consistent Ensemble Distillation, Xiaomi) - SOTA-tier audio-tagging models that classify everyday sounds (baby cry, footsteps, glass breaking, alarms, dog bark,...) into the 527-class AudioSet ontology. These files run with ced.cpp, a standalone C++/ggml port (no Python, no PyTorch at inference), and with LocalAI via the ced backend. Converted from the mispeech/ced- checkpoints (Apache-2.0). CED is a plain AST/DeiT Vision Transformer over a log-mel spectrogram; the port is numerically equal to the PyTorch reference. One self-contained GGUF per size + quant (config, 527 labels, and the mel filterbank/window are all embedded). Pick by your accuracy/size…
Open-weight model · Audio classification
wav2vec2-base-finetuned-amd
by Dmitry justin1983/wav2vec2-base-finetuned-amd
This model is a fine-tuned version of facebook/wav2vec2-base on the None dataset.
Model Card
By Dmitry, published under apache-2.0, revision 7f4b768696a4.
This model is a fine-tuned version of facebook/wav2vec2-base on the None dataset. It achieves the following results on the evaluation set: The following hyperparameters were used during training: - learningrate: 5e-05 - trainbatchsize: 32 - evalbatchsize: 32 - lrschedulertype: linear - numepochs: 2 - Transformers 4.28.0 - Pytorch 2.0.0 - Datasets 2.12.0 - Tokenizers 0.13.3
Read Dmitry's full model card
This model is a fine-tuned version of facebook/wav2vec2-base on the None dataset. It achieves the following results on the evaluation set: - Loss: 0.2777 - Accuracy: 0.8455
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training: - learning_rate: 5e-05 - train_batch_size: 32 - eval_batch_size: 32 - seed: 42 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - num_epochs: 2
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.368 | 1.0 | 204 | 0.2701 | 0.844 |
| 0.2867 | 2.0 | 408 | 0.2777 | 0.8455 |
Framework versions
- Transformers 4.28.0
- Pytorch 2.0.0
- Datasets 2.12.0
- Tokenizers 0.13.3
Configuration
- Architecture
- Wav2Vec2ForSequenceClassification
- Layers
- 12
- Hidden size
- 768
- Feed-forward size
- 3,072
- Attention heads
- 12
- Vocabulary size
- 32
- Stored precision
- float32
- Model type
- wav2vec2
Identity and Version
- Repository
- justin1983/wav2vec2-base-finetuned-amd
- Publisher
- Dmitry
- Task
- Audio classification
- Modality
- Audio
- Library
- transformers
- Parameters
- Not stated by the source
- Languages
- Not stated by the source
- Revision
- 7f4b768696a440cbcce5331bc54c4e9281291266
- First published
- 2023-05-05
- Last updated
- 2023-06-02
Files and Weights
16 files, 1.5 GB in total. The weights are 7 files totalling 1.5 GB in bin, pt, pth.
Every file
| File | Type | Size | SHA-256 |
|---|---|---|---|
| last-checkpoint/optimizer.pt | Weights | 756.7 MB | 92d9b410a83e |
| last-checkpoint/pytorch_model.bin | Weights | 378.3 MB | ebe6005ae347 |
| last-checkpoint/rng_state.pth | Weights | 13.6 KB | aa3afdb73555 |
| last-checkpoint/scheduler.pt | Weights | 627 B | 20b9fa9ec6f4 |
| last-checkpoint/training_args.bin | Weights | 3.6 KB | 0cfb16968429 |
| pytorch_model.bin | Weights | 378.3 MB | ebe6005ae347 |
| training_args.bin | Weights | 3.6 KB | 0cfb16968429 |
| config.json | Configuration | 2.5 KB | — |
| last-checkpoint/config.json | Configuration | 2.5 KB | — |
| last-checkpoint/preprocessor_config.json | Configuration | 215 B | — |
| last-checkpoint/trainer_state.json | Configuration | 7.8 KB | — |
| preprocessor_config.json | Configuration | 215 B | — |
| README.md | Documentation | 1.4 KB | — |
| .DS_Store | Repository | 6.1 KB | 93d313ecc37a |
| .gitattributes | Repository | 1.5 KB | — |
| .gitignore | Repository | 13 B | — |
License and Download
- License
- apache-2.0
- Access
- Open weights, no gate
- Download size
- 1.5 GB
Released by Dmitry through its official repository on Hugging Face. Read the license.
Memory Requirements
| Precision | Weights in memory |
|---|---|
| As published | 1.5 GB |
Weights only, from the published parameter count; the key-value cache and runtime add to this.
Questions About wav2vec2-base-finetuned-amd
Can I use wav2vec2-base-finetuned-amd commercially?
Yes. wav2vec2-base-finetuned-amd is released under Apache License 2.0. The Apache License 2.0 is a permissive open-source license. It permits commercial use, modification and redistribution. It requires keeping the license and copyright notices and any NOTICE file, stating significant changes, and it includes an express patent grant from contributors.
Similar Models
This is a spoken language recognition model trained on the VoxLingua107 dataset using SpeechBrain. The model uses the ECAPA-TDNN architecture that has previously been used for speaker recognition. However, it uses more fully connected hidden layers after the embedding layer, and cross-entropy loss was used for training. We observed that this improved the performance of extracted utterance embeddings for downstream tasks. The system is trained with recordings sampled at 16kHz (single channel). The code will automatically normalize your audio (i.e., resampling + mono channel selection) when calling classifyfile if needed. The model can classify a speech utterance according to the language…
Abstract: The recognition of accented speech still remains a dominant problem in Automatic Speech Recognition (ASR) systems. We approach the classification of accented English speech through the Emphasized Channel Attention, Propagation and Aggregation Time Delay Neural Network (ECAPA-TDNN) architecture which has been shown to perform well on a variety of speech tasks. Three models are proposed: one trained from scratch, another two models (one using data augmentation and a baseline model) fine-tuned from the checkpoints of speechbrain/spkrec-ecapa-voxceleb (VoxCeleb). Our results show that the model fine-tuned with data augmentation yield the best results. Most of the misclassifications…
The development log of our Music Audio Pre-training (m-a-p) model family: - 02/06/2023: arxiv pre-print and training codes released. - 17/03/2023: we release two advanced music understanding models, MERT-v1-95M and MERT-v1-330M, trained with new paradigm and dataset. They outperform the previous models and can better generalize to more tasks. - 14/03/2023: we retrained the MERT-v0 model with open-source-only music dataset MERT-v0-public - 29/12/2022: a music understanding model MERT-v0 trained with MLM paradigm, which performs better at downstream tasks. - 29/10/2022: a pre-trained MIR model music2vec trained with BYOL paradigm. Here is a table for quick model pick-up: The m-a-p models…
The development log of our Music Audio Pre-training (m-a-p) model family: - 02/06/2023: arxiv pre-print and training codes released. - 17/03/2023: we release two advanced music understanding models, MERT-v1-95M and MERT-v1-330M, trained with new paradigm and dataset. They outperform the previous models and can better generalize to more tasks. - 14/03/2023: we retrained the MERT-v0 model with open-source-only music dataset MERT-v0-public - 29/12/2022: a music understanding model MERT-v0 trained with MLM paradigm, which performs better at downstream tasks. - 29/10/2022: a pre-trained MIR model music2vec trained with BYOL paradigm. Here is a table for quick model pick-up: The m-a-p models…
This model combines the SpeechBrain ECAPA-TDNN speaker embedding model with an SVM classifier to predict speaker gender from audio input. The model was trained and evaluated on the VoxCeleb2, Mozilla Common Voice v10.0, and TIMIT datasets - Mozilla Common Voice v10.0 English validated test set: 92.3% accuracy The model was trained on VoxCeleb2 dataset: - Converted to WAV format, single channel, 16kHz sampling rate, 256 kp/s bitrate - Applied SileroVAD for voice activity detection, taking the first voiced segment You can install the package directly from GitHub: - Model was trained on celebrity voices from YouTube interviews - Performance may vary on different audio qualities or recording…
