Run ready

config.json

@@ -25,5 +25,5 @@
   "checkpoint": false,
   "save_step": 69,
   "data_path": "/run/shm/erogol/LJSpeech-1.0",
-  "output_path": "result",
+  "output_path": "result"
 }

datasets/LJSpeech.py

@@ -18,7 +18,6 @@ class LJSpeechDataset(Dataset):
         
         with open(csv_file, "r") as f:
             self.frames = [line.split('|') for line in f]
-            self.frames = self.frames[:256]
         self.root_dir = root_dir
         self.outputs_per_step = outputs_per_step
         self.sample_rate = sample_rate

train.py

@@ -285,6 +285,7 @@ def evaluate(model, criterion, data_loader, current_step):
         
 def main(args):
 
+    # Setup the dataset
     train_dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata_train.csv'),
                               os.path.join(c.data_path, 'wavs'),
                               c.r,
@@ -325,6 +326,7 @@ def main(args):
                             drop_last=True, num_workers= 4,
                             pin_memory=True)
 
+    model = Tacotron(c.embedding_size,
                      c.hidden_size,
                      c.num_mels,
                      c.num_freq,