Lollikit commited on
Commit
5e1d30c
·
1 Parent(s): 34aa7ff

Add model

Browse files
Files changed (3) hide show
  1. models/2.0.conf +37 -0
  2. models/2.0.pt +3 -0
  3. models/__init__.py +0 -0
models/2.0.conf ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "Model": {
3
+ "module": "transkun.ModelTransformer",
4
+ "configClassName": "Config",
5
+ "config": {
6
+ "f_min": 30,
7
+ "f_max": 8000,
8
+ "n_mels": 229,
9
+ "segmentHopSizeInSecond": 8,
10
+ "segmentSizeInSecond": 16,
11
+ "hopSize": 1024,
12
+ "windowSize": 4096,
13
+ "fs": 44100,
14
+ "nExtraWins": 5,
15
+ "baseSize": 64,
16
+ "downsampleF": true,
17
+ "posEmbedInitGamma": 1,
18
+ "nHead": 8,
19
+ "fourierSize": 64,
20
+ "nLayers": 6,
21
+ "enabledAttn": [
22
+ "F",
23
+ "T"
24
+ ],
25
+ "hiddenFactorAttn": 1,
26
+ "hiddenFactor": 4,
27
+ "velocityPredictorHiddenSize": 512,
28
+ "refinedOFPredictorHiddenSize": 512,
29
+ "scoringExpansionFactor": 4,
30
+ "useInnerProductScorer": true,
31
+ "scoreDropoutProb": 0.1,
32
+ "contextDropoutProb": 0.0,
33
+ "velocityDropoutProb": 0.1,
34
+ "refinedOFDropoutProb": 0.1
35
+ }
36
+ }
37
+ }
models/2.0.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50a80010effc2a59ffcd068a95cd2b29bd7f23a27a3515bc3ccd209c89a3d44c
3
+ size 56408978
models/__init__.py ADDED
File without changes