test(xtts): use 4 dataloader workers to match github ci

idiap · Nov 6, 2024 · 18d135e · 18d135e
1 parent ee4dae0
commit 18d135e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/xtts_tests/test_xtts_gpt_train.py b/tests/xtts_tests/test_xtts_gpt_train.py
@@ -105,7 +105,7 @@
     batch_size=BATCH_SIZE,
     batch_group_size=48,
     eval_batch_size=BATCH_SIZE,
-    num_loader_workers=8,
+    num_loader_workers=4,
     eval_split_max_size=256,
     print_step=50,
     plot_step=100,

diff --git a/tests/xtts_tests/test_xtts_v2-0_gpt_train.py b/tests/xtts_tests/test_xtts_v2-0_gpt_train.py
@@ -105,7 +105,7 @@
     batch_size=BATCH_SIZE,
     batch_group_size=48,
     eval_batch_size=BATCH_SIZE,
-    num_loader_workers=8,
+    num_loader_workers=4,
     eval_split_max_size=256,
     print_step=50,
     plot_step=100,