From 7e02cff924ba3ebc377899555f68247d22baea49 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 12 May 2021 16:10:08 +0200
Subject: [PATCH] reduce pwgan test batch size

---
 tests/vocoder_tests/test_parallel_wavegan_train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/vocoder_tests/test_parallel_wavegan_train.py b/tests/vocoder_tests/test_parallel_wavegan_train.py
index 97d3c5f1..fb6ea87c 100644
--- a/tests/vocoder_tests/test_parallel_wavegan_train.py
+++ b/tests/vocoder_tests/test_parallel_wavegan_train.py
@@ -9,14 +9,14 @@ config_path = os.path.join(get_tests_output_path(), "test_vocoder_config.json")
 output_path = os.path.join(get_tests_output_path(), "train_outputs")
 
 config = ParallelWaveganConfig(
-    batch_size=8,
-    eval_batch_size=8,
+    batch_size=4,
+    eval_batch_size=4,
     num_loader_workers=0,
     num_val_loader_workers=0,
     run_eval=True,
     test_delay_epochs=-1,
     epochs=1,
-    seq_len=8192,
+    seq_len=2048,
     eval_split_size=1,
     print_step=1,
     print_eval=True,