Testing update

tests/layers_tests.py

@@ -19,14 +19,21 @@ class PrenetTests(unittest.TestCase):
 
 class CBHGTests(unittest.TestCase):
     def test_in_out(self):
-        layer = CBHG(128, K=6, projections=[128, 128], num_highways=2)
+        layer = self.cbhg = CBHG(
+            128,
+            K=8,
+            conv_bank_features=80,
+            conv_projections=[160, 128],
+            highway_features=80,
+            gru_features=80,
+            num_highways=4)
         dummy_input = T.rand(4, 8, 128)
 
         print(layer)
         output = layer(dummy_input)
         assert output.shape[0] == 4
         assert output.shape[1] == 8
-        assert output.shape[2] == 256
+        assert output.shape[2] == 160
 
 
 class DecoderTests(unittest.TestCase):

tests/loader_tests.py

@@ -9,6 +9,8 @@ from TTS.datasets import LJSpeech, Kusal
 
 file_path = os.path.dirname(os.path.realpath(__file__))
 c = load_config(os.path.join(file_path, 'test_config.json'))
+ok_kusal = os.path.exists(c.data_path_Kusal)
+ok_ljspeech = os.path.exists(c.data_path_LJSpeech)
 
 
 class TestLJSpeechDataset(unittest.TestCase):

tests/test_config.json

@@ -29,8 +29,8 @@
     "num_loader_workers": 4,
 
     "save_step": 200,
-    "data_path_LJSpeech": "C:/Users/erogol/Data/LJSpeech-1.1",
-    "data_path_Kusal": "C:/Users/erogol/Data/Kusal",
+    "data_path_LJSpeech": "/home/erogol/Data/LJSpeech-1.1",
+    "data_path_Kusal": "/home/erogol/Data/Kusal",
     "output_path": "result",
     "min_seq_len": 0,
     "log_dir": "/home/erogol/projects/TTS/logs/"

train.py

@@ -24,7 +24,7 @@ from utils.audio import AudioProcessor
 
 
 torch.manual_seed(1)
-torch.set_num_threads(4)
+# torch.set_num_threads(4)
 use_cuda = torch.cuda.is_available()