Christian J. Steinmetz commited on
Commit
8a6eea8
1 Parent(s): 279d167

cleaning up fx classifiers

Browse files
cfg/model/cls_panns_16k.yaml CHANGED
@@ -10,6 +10,6 @@ model:
10
  n_fft: 2048
11
  hop_length: 512
12
  n_mels: 128
13
- sample_rate: 44100
14
- model_sample_rate: 16000
15
 
 
10
  n_fft: 2048
11
  hop_length: 512
12
  n_mels: 128
13
+ sample_rate: ${sample_rate}
14
+ model_sample_rate: ${sample_rate}
15
 
cfg/model/cls_panns_44k.yaml CHANGED
@@ -11,7 +11,7 @@ model:
11
  n_fft: 2048
12
  hop_length: 512
13
  n_mels: 128
14
- sample_rate: 44100
15
- model_sample_rate: 44100
16
  specaugment: False
17
 
 
11
  n_fft: 2048
12
  hop_length: 512
13
  n_mels: 128
14
+ sample_rate: ${sample_rate}
15
+ model_sample_rate: ${sample_rate}
16
  specaugment: False
17
 
cfg/model/{cls_panns_16k_mixup.yaml → cls_panns_44k_label_smoothing.yaml} RENAMED
@@ -5,12 +5,13 @@ model:
5
  lr_weight_decay: 1e-3
6
  sample_rate: ${sample_rate}
7
  mixup: True
 
8
  network:
9
  _target_: remfx.classifier.Cnn14
10
  num_classes: ${num_classes}
11
  n_fft: 2048
12
  hop_length: 512
13
  n_mels: 128
14
- sample_rate: 44100
15
- model_sample_rate: 16000
16
-
 
5
  lr_weight_decay: 1e-3
6
  sample_rate: ${sample_rate}
7
  mixup: True
8
+ label_smoothing: 0.1
9
  network:
10
  _target_: remfx.classifier.Cnn14
11
  num_classes: ${num_classes}
12
  n_fft: 2048
13
  hop_length: 512
14
  n_mels: 128
15
+ sample_rate: ${sample_rate}
16
+ model_sample_rate: ${sample_rate}
17
+ specaugment: False
cfg/model/cls_panns_44k_mixup.yaml CHANGED
@@ -10,8 +10,7 @@ model:
10
  num_classes: ${num_classes}
11
  n_fft: 2048
12
  hop_length: 512
13
- n_mels: 64
14
- sample_rate: 44100
15
- model_sample_rate: 44100
16
  specaugment: False
17
-
 
10
  num_classes: ${num_classes}
11
  n_fft: 2048
12
  hop_length: 512
13
+ n_mels: 128
14
+ sample_rate: ${sample_rate}
15
+ model_sample_rate: ${sample_rate}
16
  specaugment: False
 
cfg/model/cls_panns_pt_mixup.yaml DELETED
@@ -1,12 +0,0 @@
1
- # @package _global_
2
- model:
3
- _target_: remfx.models.FXClassifier
4
- lr: 3e-4
5
- lr_weight_decay: 1e-3
6
- sample_rate: ${sample_rate}
7
- mixup: True
8
- network:
9
- _target_: remfx.classifier.PANNs
10
- num_classes: ${num_classes}
11
- sample_rate: ${sample_rate}
12
-