reduce pwgan test batch size

This commit is contained in:
Eren Gölge 2021-05-12 16:10:08 +02:00
parent da49089a72
commit 7e02cff924
1 changed files with 3 additions and 3 deletions

View File

@ -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") output_path = os.path.join(get_tests_output_path(), "train_outputs")
config = ParallelWaveganConfig( config = ParallelWaveganConfig(
batch_size=8, batch_size=4,
eval_batch_size=8, eval_batch_size=4,
num_loader_workers=0, num_loader_workers=0,
num_val_loader_workers=0, num_val_loader_workers=0,
run_eval=True, run_eval=True,
test_delay_epochs=-1, test_delay_epochs=-1,
epochs=1, epochs=1,
seq_len=8192, seq_len=2048,
eval_split_size=1, eval_split_size=1,
print_step=1, print_step=1,
print_eval=True, print_eval=True,