Update config

config/LibriTTS/ar.yml

@@ -0,0 +1,5 @@
+data_dirs: [data/LibriTTS/train-clean-100/]
+spkr_name_getter: "lambda p: p.parts[-3]"
+
+model: ar-quarter
+batch_size: 8

config/LibriTTS/nar.yml

@@ -0,0 +1,5 @@
+data_dirs: [data/LibriTTS/train-clean-100/]
+spkr_name_getter: "lambda p: p.parts[-3]"
+
+model: nar-quarter
+batch_size: 8

config/test/ar.yml

@@ -1,4 +1,4 @@
 data_dirs: [data/test]
 
-model: ar
+model: ar-quarter
 batch_size: 1

config/test/nar.yml

@@ -1,4 +1,4 @@
 data_dirs: [data/test]
 
-model: nar
+model: nar-quarter
 batch_size: 1

vall_e/config.py

@@ -23,8 +23,8 @@ class Config(ConfigBase):
     num_tokens: int = 1024
 
     nj: int = 8
-    batch_size: int = 128
-    eval_batch_size: int = 512
+    batch_size: int = 32
+    eval_batch_size: int = 32
     warmup_min_lr: float = 1e-6
     warmup_max_lr: float = 2e-4
     dis_warmup_max_lr: float = 4e-4