|
|
--- |
|
|
tags: |
|
|
- espnet |
|
|
- audio |
|
|
- classification |
|
|
datasets: |
|
|
- esc50 |
|
|
license: cc-by-4.0 |
|
|
--- |
|
|
|
|
|
## ESPnet2 CLS model |
|
|
|
|
|
### `shikhar7ssu/OpenBEATS-Large-i1-esc50f5` |
|
|
|
|
|
This model was trained by Shikhar Bharadwaj using esc50 recipe in [espnet](https://github.com/espnet/espnet/). |
|
|
|
|
|
## CLS config |
|
|
|
|
|
<details><summary>expand</summary> |
|
|
|
|
|
``` |
|
|
config: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/earlarge1/conf/ear_large/esc50_f5.yaml |
|
|
print_config: false |
|
|
log_level: INFO |
|
|
drop_last_iter: false |
|
|
dry_run: false |
|
|
iterator_type: sequence |
|
|
valid_iterator_type: null |
|
|
output_dir: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/esc50_f5/cls_earlarge1 |
|
|
ngpu: 1 |
|
|
seed: 0 |
|
|
num_workers: 2 |
|
|
num_att_plot: 0 |
|
|
dist_backend: nccl |
|
|
dist_init_method: env:// |
|
|
dist_world_size: null |
|
|
dist_rank: null |
|
|
local_rank: 0 |
|
|
dist_master_addr: null |
|
|
dist_master_port: null |
|
|
dist_launcher: null |
|
|
multiprocessing_distributed: false |
|
|
unused_parameters: true |
|
|
sharded_ddp: false |
|
|
use_deepspeed: false |
|
|
deepspeed_config: null |
|
|
gradient_as_bucket_view: true |
|
|
ddp_comm_hook: null |
|
|
cudnn_enabled: true |
|
|
cudnn_benchmark: false |
|
|
cudnn_deterministic: true |
|
|
use_tf32: false |
|
|
collect_stats: false |
|
|
write_collected_feats: false |
|
|
max_epoch: 1000 |
|
|
patience: null |
|
|
val_scheduler_criterion: |
|
|
- valid |
|
|
- loss |
|
|
early_stopping_criterion: |
|
|
- valid |
|
|
- loss |
|
|
- min |
|
|
best_model_criterion: |
|
|
- - valid |
|
|
- acc |
|
|
- max |
|
|
keep_nbest_models: 1 |
|
|
nbest_averaging_interval: 0 |
|
|
grad_clip: 1 |
|
|
grad_clip_type: 2.0 |
|
|
grad_noise: false |
|
|
accum_grad: 1 |
|
|
no_forward_run: false |
|
|
resume: true |
|
|
train_dtype: float32 |
|
|
use_amp: false |
|
|
log_interval: null |
|
|
use_matplotlib: true |
|
|
use_tensorboard: true |
|
|
create_graph_in_tensorboard: false |
|
|
use_wandb: false |
|
|
wandb_project: null |
|
|
wandb_id: null |
|
|
wandb_entity: null |
|
|
wandb_name: null |
|
|
wandb_model_log_interval: -1 |
|
|
detect_anomaly: false |
|
|
use_adapter: false |
|
|
adapter: lora |
|
|
save_strategy: all |
|
|
adapter_conf: {} |
|
|
pretrain_path: null |
|
|
init_param: [] |
|
|
ignore_init_mismatch: false |
|
|
freeze_param: [] |
|
|
num_iters_per_epoch: null |
|
|
batch_size: 512 |
|
|
valid_batch_size: null |
|
|
batch_bins: 1000000 |
|
|
valid_batch_bins: null |
|
|
category_sample_size: 10 |
|
|
train_shape_file: |
|
|
- /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/esc50_f5/cls_stats_16k/train/speech_shape |
|
|
- /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/esc50_f5/cls_stats_16k/train/label_shape |
|
|
valid_shape_file: |
|
|
- /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/esc50_f5/cls_stats_16k/valid/speech_shape |
|
|
- /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/esc50_f5/cls_stats_16k/valid/label_shape |
|
|
batch_type: folded |
|
|
valid_batch_type: null |
|
|
fold_length: |
|
|
- 100000 |
|
|
- 1 |
|
|
sort_in_batch: descending |
|
|
shuffle_within_batch: false |
|
|
sort_batch: descending |
|
|
multiple_iterator: false |
|
|
chunk_length: 500 |
|
|
chunk_shift_ratio: 0.5 |
|
|
num_cache_chunks: 1024 |
|
|
chunk_excluded_key_prefixes: [] |
|
|
chunk_default_fs: null |
|
|
chunk_max_abs_length: null |
|
|
chunk_discard_short_samples: true |
|
|
train_data_path_and_name_and_type: |
|
|
- - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/esc50_f5/train5/wav.scp |
|
|
- speech |
|
|
- sound |
|
|
- - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/esc50_f5/train5/text |
|
|
- label |
|
|
- text |
|
|
valid_data_path_and_name_and_type: |
|
|
- - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/esc50_f5/val5/wav.scp |
|
|
- speech |
|
|
- sound |
|
|
- - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/esc50_f5/val5/text |
|
|
- label |
|
|
- text |
|
|
multi_task_dataset: false |
|
|
allow_variable_data_keys: false |
|
|
max_cache_size: 0.0 |
|
|
max_cache_fd: 32 |
|
|
allow_multi_rates: false |
|
|
valid_max_cache_size: null |
|
|
exclude_weight_decay: false |
|
|
exclude_weight_decay_conf: {} |
|
|
optim: adamw |
|
|
optim_conf: |
|
|
lr: 0.0001 |
|
|
weight_decay: 0.01 |
|
|
betas: |
|
|
- 0.9 |
|
|
- 0.98 |
|
|
scheduler: cosineannealingwarmuprestarts |
|
|
scheduler_conf: |
|
|
first_cycle_steps: 7000 |
|
|
warmup_steps: 350 |
|
|
max_lr: 0.0001 |
|
|
min_lr: 5.0e-06 |
|
|
token_list: |
|
|
- audio_class:0 |
|
|
- audio_class:14 |
|
|
- audio_class:36 |
|
|
- audio_class:19 |
|
|
- audio_class:30 |
|
|
- audio_class:34 |
|
|
- audio_class:9 |
|
|
- audio_class:22 |
|
|
- audio_class:48 |
|
|
- audio_class:41 |
|
|
- audio_class:47 |
|
|
- audio_class:31 |
|
|
- audio_class:17 |
|
|
- audio_class:45 |
|
|
- audio_class:8 |
|
|
- audio_class:15 |
|
|
- audio_class:46 |
|
|
- audio_class:37 |
|
|
- audio_class:32 |
|
|
- audio_class:16 |
|
|
- audio_class:25 |
|
|
- audio_class:4 |
|
|
- audio_class:3 |
|
|
- audio_class:27 |
|
|
- audio_class:43 |
|
|
- audio_class:12 |
|
|
- audio_class:40 |
|
|
- audio_class:29 |
|
|
- audio_class:10 |
|
|
- audio_class:7 |
|
|
- audio_class:26 |
|
|
- audio_class:6 |
|
|
- audio_class:44 |
|
|
- audio_class:23 |
|
|
- audio_class:20 |
|
|
- audio_class:49 |
|
|
- audio_class:24 |
|
|
- audio_class:39 |
|
|
- audio_class:28 |
|
|
- audio_class:18 |
|
|
- audio_class:2 |
|
|
- audio_class:35 |
|
|
- audio_class:38 |
|
|
- audio_class:21 |
|
|
- audio_class:1 |
|
|
- audio_class:11 |
|
|
- audio_class:42 |
|
|
- audio_class:5 |
|
|
- audio_class:33 |
|
|
- audio_class:13 |
|
|
- <blank> |
|
|
- <unk> |
|
|
text_token_list: null |
|
|
text_bpemodel: null |
|
|
init: xavier_normal |
|
|
input_size: 1 |
|
|
use_preprocessor: true |
|
|
frontend: null |
|
|
frontend_conf: {} |
|
|
specaug: null |
|
|
specaug_conf: {} |
|
|
normalize: null |
|
|
normalize_conf: {} |
|
|
preencoder: null |
|
|
preencoder_conf: {} |
|
|
encoder: beats |
|
|
encoder_conf: |
|
|
beats_ckpt_path: /work/nvme/bbjs/sbharadwaj/model_checkpoints/ear_large/beats_iter0_large.tune_lr1.0e-4_warmup40000_bins1600000_totalsteps400000/epoch59.pt |
|
|
fbank_mean: 11.72215 |
|
|
fbank_std: 10.60431 |
|
|
beats_config: |
|
|
layer_wise_gradient_decay_ratio: 0.2 |
|
|
encoder_layerdrop: 0.1 |
|
|
dropout: 0.0 |
|
|
specaug_config: |
|
|
apply_time_warp: true |
|
|
apply_freq_mask: false |
|
|
apply_time_mask: true |
|
|
time_mask_width_ratio_range: |
|
|
- 0 |
|
|
- 0.06 |
|
|
num_time_mask: 1 |
|
|
roll_augment: true |
|
|
roll_interval: 16000 |
|
|
use_weighted_representation: false |
|
|
text_encoder: null |
|
|
text_encoder_conf: {} |
|
|
embedding_fusion: null |
|
|
embedding_fusion_conf: {} |
|
|
decoder: linear |
|
|
decoder_conf: |
|
|
pooling: mean |
|
|
dropout: 0.1 |
|
|
model: espnet |
|
|
model_conf: |
|
|
classification_type: multi-class |
|
|
lsm_weight: 0.1 |
|
|
required: |
|
|
- output_dir |
|
|
- token_list |
|
|
version: '202412' |
|
|
distributed: false |
|
|
``` |
|
|
|
|
|
</details> |
|
|
|
|
|
### Citations |
|
|
|
|
|
```BibTex |
|
|
|
|
|
@article{bharadwaj2025openbeats, |
|
|
title={OpenBEATs: A Fully Open-Source General-Purpose Audio Encoder}, |
|
|
author={Bharadwaj, Shikhar and Cornell, Samuele and Choi, Kwanghee and Fukayama, Satoru and Shim, Hye-jin and Deshmukh, Soham and Watanabe, Shinji}, |
|
|
journal={arXiv preprint arXiv:2507.14129}, |
|
|
year={2025} |
|
|
} |
|
|
|
|
|
@inproceedings{watanabe2018espnet, |
|
|
author={Shinji Watanabe and Takaaki Hori and Shigeki Karita and Tomoki Hayashi and Jiro Nishitoba and Yuya Unno and Nelson Yalta and Jahn Heymann and Matthew Wiesner and Nanxin Chen and Adithya Renduchintala and Tsubasa Ochiai}, |
|
|
title={{ESPnet}: End-to-End Speech Processing Toolkit}, |
|
|
year={2018}, |
|
|
booktitle={Proceedings of Interspeech}, |
|
|
pages={2207--2211}, |
|
|
doi={10.21437/Interspeech.2018-1456}, |
|
|
url={http://dx.doi.org/10.21437/Interspeech.2018-1456} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
``` |
|
|
|
