Shikhar Bharadwaj commited on
Commit
3f7aefe
·
1 Parent(s): 5b0b70f

Update model

Browse files
meta.yaml CHANGED
@@ -2,7 +2,7 @@ espnet: '202503'
2
  files:
3
  classification_model_file: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/valid.epoch_mAP.ave_1best.pth
4
  python: "3.9.18 | packaged by conda-forge | (main, Dec 23 2023, 17:20:25) \n[GCC 12.3.0]"
5
- timestamp: 1763330218.090096
6
  torch: 2.1.2
7
  yaml_files:
8
  classification_train_config: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/config.yaml
 
2
  files:
3
  classification_model_file: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/valid.epoch_mAP.ave_1best.pth
4
  python: "3.9.18 | packaged by conda-forge | (main, Dec 23 2023, 17:20:25) \n[GCC 12.3.0]"
5
+ timestamp: 1763330253.214344
6
  torch: 2.1.2
7
  yaml_files:
8
  classification_train_config: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/config.yaml
work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/lightning_logs/version_0/events.out.tfevents.1746445643.gh132.hsn.cm.delta.internal.ncsa.edu.3804512.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6fe910543e2907bf59e7d16e8af07e973e3b0a28dd7416442b21156ad86128c
3
+ size 282738
work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3/lightning_logs/version_0/hparams.yaml ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ args: !!python/object:argparse.Namespace
2
+ accum_grad: 2
3
+ adapter: lora
4
+ adapter_conf: {}
5
+ allow_multi_rates: false
6
+ allow_variable_data_keys: false
7
+ batch_bins: 20000000
8
+ batch_size: 20
9
+ batch_type: length_weighted
10
+ best_model_criterion:
11
+ - - valid
12
+ - epoch_mAP
13
+ - max
14
+ category_sample_size: 10
15
+ chunk_default_fs: null
16
+ chunk_discard_short_samples: true
17
+ chunk_excluded_key_prefixes: []
18
+ chunk_length: 500
19
+ chunk_max_abs_length: null
20
+ chunk_shift_ratio: 0.5
21
+ collect_stats: false
22
+ config: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/earlarge3.final/conf/ear_large/audioset2m.yaml
23
+ create_graph_in_tensorboard: false
24
+ cudnn_benchmark: false
25
+ cudnn_deterministic: true
26
+ cudnn_enabled: true
27
+ ddp_comm_hook: null
28
+ decoder: linear
29
+ decoder_conf: {}
30
+ deepspeed_config: null
31
+ detect_anomaly: false
32
+ dist_backend: nccl
33
+ dist_init_method: env://
34
+ dist_launcher: null
35
+ dist_master_addr: null
36
+ dist_master_port: null
37
+ dist_rank: null
38
+ dist_world_size: null
39
+ drop_last_iter: false
40
+ dry_run: false
41
+ early_stopping_criterion: !!python/tuple
42
+ - valid
43
+ - loss
44
+ - min
45
+ embedding_fusion: null
46
+ embedding_fusion_conf: {}
47
+ encoder: beats
48
+ encoder_conf:
49
+ beats_ckpt_path: /work/nvme/bbjs/sbharadwaj/7Msounds/exp/beats_iter2_large2.tune_lr1.0e-4_warmup40000_bins1600000_totalsteps400000/epoch_latest.pt
50
+ beats_config:
51
+ dropout: 0.0
52
+ encoder_layerdrop: 0.1
53
+ layer_wise_gradient_decay_ratio: 0.6
54
+ roll_augment: true
55
+ roll_interval: 1
56
+ specaug_config:
57
+ apply_freq_mask: false
58
+ apply_time_mask: true
59
+ apply_time_warp: true
60
+ num_time_mask: 1
61
+ time_mask_width_ratio_range:
62
+ - 0
63
+ - 0.06
64
+ use_weighted_representation: false
65
+ exclude_weight_decay: false
66
+ exclude_weight_decay_conf: {}
67
+ fold_length:
68
+ - 160000
69
+ - 600
70
+ freeze_param: []
71
+ frontend: null
72
+ frontend_conf:
73
+ fs: 16k
74
+ grad_clip: 1
75
+ grad_clip_type: 2.0
76
+ grad_noise: false
77
+ gradient_as_bucket_view: true
78
+ ignore_init_mismatch: false
79
+ init: xavier_normal
80
+ init_param: []
81
+ input_size: 1
82
+ iterator_type: sequence
83
+ keep_nbest_models: 1
84
+ lightning_conf:
85
+ best_model_criterion:
86
+ - - valid/epoch_mAP
87
+ - max
88
+ - 1
89
+ default_root_dir: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3.final
90
+ devices: 4
91
+ log_every_n_steps: 250
92
+ max_epochs: 25
93
+ num_nodes: 1
94
+ strategy: ddp
95
+ strategy_conf:
96
+ find_unused_parameters: true
97
+ local_rank: null
98
+ log_interval: null
99
+ log_level: INFO
100
+ max_cache_fd: 32
101
+ max_cache_size: 0.0
102
+ max_epoch: 25
103
+ model: espnet
104
+ model_conf:
105
+ classification_type: multi-label
106
+ log_epoch_metrics: true
107
+ lsm_weight: 0.1
108
+ mixup_probability: 0.8
109
+ multi_task_dataset: false
110
+ multiple_iterator: false
111
+ multiprocessing_distributed: false
112
+ nbest_averaging_interval: 0
113
+ ngpu: 0
114
+ no_forward_run: false
115
+ normalize: null
116
+ normalize_conf: {}
117
+ num_att_plot: 0
118
+ num_cache_chunks: 1024
119
+ num_iters_per_epoch: null
120
+ num_workers: 2
121
+ optim: adamw
122
+ optim_conf:
123
+ betas:
124
+ - 0.9
125
+ - 0.98
126
+ lr: 0.0001
127
+ weight_decay: 0.01
128
+ output_dir: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_earlarge3.final
129
+ patience: null
130
+ preencoder: null
131
+ preencoder_conf: {}
132
+ pretrain_path: null
133
+ print_config: false
134
+ required:
135
+ - output_dir
136
+ - token_list
137
+ resume: true
138
+ save_strategy: all
139
+ scheduler: cosineannealingwarmuprestarts
140
+ scheduler_conf:
141
+ first_cycle_steps: 400000
142
+ max_lr: 0.0001
143
+ min_lr: 5.0e-06
144
+ warmup_steps: 10000
145
+ seed: 0
146
+ sharded_ddp: false
147
+ shuffle_within_batch: false
148
+ sort_batch: descending
149
+ sort_in_batch: descending
150
+ specaug: null
151
+ specaug_conf: {}
152
+ task: cls
153
+ text_bpemodel: null
154
+ text_encoder: null
155
+ text_encoder_conf: {}
156
+ text_token_list: null
157
+ token_list: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/data/audioset2m/token_list
158
+ train_data_path_and_name_and_type:
159
+ - !!python/tuple
160
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/audioset2m/train/wav.scp
161
+ - speech
162
+ - sound
163
+ - !!python/tuple
164
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/audioset2m/train/text
165
+ - label
166
+ - text
167
+ train_dtype: float32
168
+ train_shape_file:
169
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_stats_16k/train/speech_shape
170
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_stats_16k/train/label_shape
171
+ unused_parameters: true
172
+ use_adapter: false
173
+ use_amp: false
174
+ use_deepspeed: false
175
+ use_matplotlib: true
176
+ use_preprocessor: true
177
+ use_tensorboard: true
178
+ use_tf32: false
179
+ use_wandb: true
180
+ user_callbacks:
181
+ - mAP_logging
182
+ utt2weight_file: /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/data/audioset2m/utt2weight
183
+ val_scheduler_criterion: !!python/tuple
184
+ - valid
185
+ - loss
186
+ valid_batch_bins: null
187
+ valid_batch_size: null
188
+ valid_batch_type: length
189
+ valid_data_path_and_name_and_type:
190
+ - !!python/tuple
191
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/audioset2m/eval/wav.scp
192
+ - speech
193
+ - sound
194
+ - !!python/tuple
195
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/dump/audioset2m/eval/text
196
+ - label
197
+ - text
198
+ valid_iterator_type: null
199
+ valid_max_cache_size: null
200
+ valid_shape_file:
201
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_stats_16k/valid/speech_shape
202
+ - /work/nvme/bbjs/sbharadwaj/espnet/egs2/audioverse/v1/exp/audioset2m/cls_stats_16k/valid/label_shape
203
+ wandb_entity: shikhar
204
+ wandb_id: null
205
+ wandb_model_log_interval: -1
206
+ wandb_name: audioset2m.earlarge3.final
207
+ wandb_project: audioverse
208
+ write_collected_feats: false