Reduce test size

should fix recent travis timeout failures

Author: r9y9

tests/test_conv.py

@@ -55,9 +55,9 @@ def __test(kernel_size, dilation, T, B, C, causual=True):
 
         assert (output_conv == output_conv_online).all()
 
-    for B in [1, 16]:
-        for T in [10, 20, 30]:
+    for B in [1, 4]:
+        for T in [5, 10]:
             for C in [1, 2, 4]:
-                for kernel_size in [2, 3, 4, 5, 6, 7, 8, 9]:
-                    for dilation in [1, 2, 3, 4, 5, 6, 7, 8, 9, 27]:
+                for kernel_size in [2, 3]:
+                    for dilation in [1, 2, 3, 4, 5, 9, 27]:
                         yield __test, kernel_size, dilation, T, B, C

tests/test_deepvoice3.py

@@ -28,7 +28,7 @@ def _get_model(n_speakers=1, speaker_embed_dim=None,
                force_monotonic_attention=False,
                use_decoder_state_for_postnet_input=False):
     model = deepvoice3(n_vocab=n_vocab,
-                       embed_dim=256,
+                       embed_dim=32,
                        mel_dim=num_mels,
                        linear_dim=num_freq,
                        r=outputs_per_step,
@@ -37,9 +37,9 @@ def _get_model(n_speakers=1, speaker_embed_dim=None,
                        speaker_embed_dim=speaker_embed_dim,
                        dropout=1 - 0.95,
                        kernel_size=5,
-                       encoder_channels=128,
-                       decoder_channels=256,
-                       converter_channels=256,
+                       encoder_channels=16,
+                       decoder_channels=32,
+                       converter_channels=32,
                        force_monotonic_attention=force_monotonic_attention,
                        use_decoder_state_for_postnet_input=use_decoder_state_for_postnet_input,
                        )