fix batch_size, gpus

3 years ago · 51206ccb8e
parent 94dbb78428
commit 51206ccb8e
5 changed files with 5 additions and 5 deletions
--- a/examples/aishell3/vits-vc/conf/default.yaml
+++ b/examples/aishell3/vits-vc/conf/default.yaml
@ -146,7 +146,7 @@ cache_generator_outputs: True # whether to cache generator outputs in the traini
 ###########################################################
 #                  DATA LOADER SETTING                    #
 ###########################################################
-batch_size: 64              # Batch size.
+batch_size: 50              # Batch size.
 num_workers: 4              # Number of workers in DataLoader.
 ##########################################################
--- a/examples/aishell3/vits-vc/run.sh
+++ b/examples/aishell3/vits-vc/run.sh
@ -3,7 +3,7 @@
 set -e
 source path.sh
-gpus=0,1
+gpus=0,1,2,3
 stage=0
 stop_stage=100
--- a/examples/aishell3/vits/conf/default.yaml
+++ b/examples/aishell3/vits/conf/default.yaml
@ -145,7 +145,7 @@ cache_generator_outputs: True # whether to cache generator outputs in the traini
 ###########################################################
 #                  DATA LOADER SETTING                    #
 ###########################################################
-batch_size: 64              # Batch size.
+batch_size: 50              # Batch size.
 num_workers: 4              # Number of workers in DataLoader.
 ##########################################################
--- a/examples/aishell3/vits/run.sh
+++ b/examples/aishell3/vits/run.sh
@ -3,7 +3,7 @@
 set -e
 source path.sh
-gpus=0,1
+gpus=0,1,2,3
 stage=0
 stop_stage=100
--- a/examples/csmsc/vits/run.sh
+++ b/examples/csmsc/vits/run.sh
@ -3,7 +3,7 @@
 set -e
 source path.sh
-gpus=0,1
+gpus=0,1,2,3
 stage=0
 stop_stage=100