upload

best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5c3898e3650f74537236520fcc152cc04010cd348771219e43737d8487a70ec
+size 5247422

last.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe79fb9d063759b39cdd9023e211efb32aa052d1d2fe9a0b838c0fa0fce1385f
+size 5247422

settings.txt

@@ -0,0 +1,69 @@
+data_root: ./data_all/
+exp:
+  cache: 2
+  device: &id001 !!python/object/apply:torch.device
+  - cuda
+  exp_dir: ./runs
+  exp_name: kw-mlp-mul-final-2048-dropout-0.2
+  log_freq: 20
+  log_to_file: true
+  log_to_stdout: true
+  n_workers: 16
+  pin_memory: true
+  proj_name: torch-kw-mlp-lc
+  save_dir: ./runs/kw-mlp-mul-final-2048-dropout-0.2
+  val_freq: 1
+  wandb: false
+  wandb_api_key: null
+hparams:
+  audio:
+    center: false
+    hop_length: 160
+    n_fft: 480
+    n_mels: 40
+    sr: 16000
+    win_length: 480
+  augment:
+    spec_aug:
+      freq_mask_width: 7
+      n_freq_masks: 1
+      n_time_masks: 1
+      time_mask_width: 20
+  batch_size: 2048
+  device: *id001
+  l_smooth: 0.1
+  model:
+    channels: 1
+    depth: 12
+    dim: 64
+    dropout: 0.2
+    input_res:
+    - 40
+    - 98
+    multitask: true
+    num_classes: 35
+    num_langs: 15
+    patch_res:
+    - 40
+    - 1
+    pre_norm: false
+    prob_survival: 0.9
+    type: kw-mlp
+  n_epochs: 1000
+  optimizer:
+    opt_kwargs:
+      lr: 0.001
+      weight_decay: 0.1
+    opt_type: adamw
+  restore_ckpt: null
+  scheduler:
+    max_epochs: 1000
+    n_warmup: 10
+    scheduler_type: cosine_annealing
+  seed: 0
+  start_epoch: 0
+label_map: ./data_all/label_map.json
+lang_map: ./data_all/lang_map.json
+test_list_file: ./data_all/testing_list.txt
+train_list_file: ./data_all/training_list.txt
+val_list_file: ./data_all/validation_list.txt