rharukawa commited on
Commit
c5825ad
·
verified ·
1 Parent(s): 4578686

Upload folder using huggingface_hub

Browse files
16k-subband-qat/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:63840420b1e196ea54f42ec8c39a0937e9df34322b642e8ff74d73e79c7567bd
3
+ size 3793092
16k-subband-qat/config.yaml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ seed: 0
2
+ device: auto
3
+ stft:
4
+ sample_rate: 16000
5
+ n_fft: 1024
6
+ hop_length: 200
7
+ win_length: 800
8
+ log_eps: 1.0e-05
9
+ model:
10
+ variant: subband
11
+ base_channels: 64
12
+ depth: 4
13
+ gru_hidden: 256
14
+ gru_reduce: 32
15
+ bands: 48
16
+ blocks: 4
17
+ head: bin
18
+ in_ch: 1
19
+ out_ch: 1
20
+ mask_type: magnitude
21
+ lookahead: 0
22
+ compress_power: 1.0
23
+ norm:
24
+ mean: -2.332935
25
+ std: 1.910467
26
+ aug:
27
+ snr_db:
28
+ - 0.0
29
+ - 20.0
30
+ reverb_prob: 0.5
31
+ noise_prob: 1.0
32
+ remove_reverb: false
33
+ data:
34
+ clean_dirs:
35
+ - data/clean_24k
36
+ noise_dir: data/noise
37
+ rir_dir: data/rir
38
+ segment_frames: 128
39
+ val_fraction: 0.05
40
+ num_workers: 16
41
+ dummy: false
42
+ dummy_size: 64
43
+ train:
44
+ batch_size: 16
45
+ lr: 2.0e-05
46
+ lr_schedule: none
47
+ warmup_steps: 0
48
+ max_steps: 10000
49
+ log_every: 100
50
+ val_every: 2000
51
+ ckpt_every: 2000
52
+ ckpt_dir: outputs/ckpt_16k_subband_c64bin_sisdr_qat_h
53
+ amp: true
54
+ si_sdr_weight: 0.05
55
+ pesq_weight: 0.3
56
+ qat: false
57
+ qat_freeze_steps: 0
58
+ qat_scales: ''
59
+ init_weights: ''
16k-subband-qat/model_int8.tflite ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e21884c9c09bacaf79370d1b8681c4400b986fb8db99d06ef03191f68d2e7ce
3
+ size 370432
16k-subband-qat/model_int8_vela_u55.tflite ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38b6b59ef264784d36cf874843af7e8dbd8ee0633934176c816f890d63058396
3
+ size 302256