Upload 2 files

config.yml

@@ -0,0 +1,30 @@
+log_dir: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/checkpoint_es"
+mixed_precision: "fp16"
+data_folder: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/spanish.es.processed_fixed"
+batch_size: 4
+save_interval: 5000
+log_interval: 10
+num_process: 1 # number of GPUs
+num_steps: 1000000
+
+dataset_params:
+    tokenizer: "transfo-xl-wt103"
+    token_separator: " " # token used for phoneme separator (space)
+    token_mask: "M" # token used for phoneme mask (M)
+    word_separator: 3039 # token used for word separator (<formula>)
+    token_maps: "/gpfs/projects/bsc88/speech/TTS/repos/PL-BERT/token_maps_es_new_fix.pkl" # token map path
+    
+    max_mel_length: 512 # max phoneme length
+    
+    word_mask_prob: 0.15 # probability to mask the entire word
+    phoneme_mask_prob: 0.1 # probability to mask each phoneme
+    replace_prob: 0.2 # probablity to replace phonemes
+    
+model_params:
+    vocab_size: 178
+    hidden_size: 768
+    num_attention_heads: 12
+    intermediate_size: 2048
+    max_position_embeddings: 512
+    num_hidden_layers: 12
+    dropout: 0.1

token_maps.pkl

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