rjzevallos commited on
Commit
e62d253
·
verified ·
1 Parent(s): 09aecff

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.yml +30 -0
  2. step_1000000.t7 +3 -0
  3. token_maps.pkl +3 -0
config.yml ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ log_dir: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/checkpoint_cat"
2
+ mixed_precision: "fp16"
3
+ data_folder: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/catalan.cat.processed_fixed"
4
+ batch_size: 4
5
+ save_interval: 5000
6
+ log_interval: 10
7
+ num_process: 1 # number of GPUs
8
+ num_steps: 1000000
9
+
10
+ dataset_params:
11
+ tokenizer: "transfo-xl-wt103"
12
+ token_separator: " " # token used for phoneme separator (space)
13
+ token_mask: "M" # token used for phoneme mask (M)
14
+ word_separator: 3039 # token used for word separator (<formula>)
15
+ token_maps: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/token_maps_cat_new_fix.pkl" # token map path
16
+
17
+ max_mel_length: 512 # max phoneme length
18
+
19
+ word_mask_prob: 0.15 # probability to mask the entire word
20
+ phoneme_mask_prob: 0.1 # probability to mask each phoneme
21
+ replace_prob: 0.2 # probablity to replace phonemes
22
+
23
+ model_params:
24
+ vocab_size: 178
25
+ hidden_size: 768
26
+ num_attention_heads: 12
27
+ intermediate_size: 2048
28
+ max_position_embeddings: 512
29
+ num_hidden_layers: 12
30
+ dropout: 0.1
step_1000000.t7 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec4ab05dbb489d1c0f9ffcfc9c9e3745ba375d4a050faaf99847df26cc19be08
3
+ size 10538483255
token_maps.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4f0f232526a5acc006e1a17808d8828d0631fcae42629245e83c3faad840a5f
3
+ size 39266574