From 18d135e4523e9d983e7f9443aee477478fa31f09 Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Mon, 21 Oct 2024 00:45:00 +0200 Subject: [PATCH] test(xtts): use 4 dataloader workers to match github ci --- tests/xtts_tests/test_xtts_gpt_train.py | 2 +- tests/xtts_tests/test_xtts_v2-0_gpt_train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/xtts_tests/test_xtts_gpt_train.py b/tests/xtts_tests/test_xtts_gpt_train.py index b8b9a4e388..29e5de3deb 100644 --- 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 index 6663433c12..da4f923706 100644 --- 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,