perf: set batch size to 64

app/setup.py

@@ -61,7 +61,8 @@ class SetupTab:
     def ui(self):
         gr.Markdown("# Setup Experiment")
         gr.Markdown(
-            "You can upload a zip file containing audio files to start a new experiment, or upload an experiment directory zip file to restore an existing experiment."
+            "You can upload a zip file containing audio files to start a new experiment, or upload an experiment directory zip file to restore an existing experiment.\n"
+            "The suggested dataset size is > 5 minutes of audio."
         )
 
         with gr.Row():

config.json

@@ -58,7 +58,7 @@
         "use_spectral_norm": false
     },
     "train": {
-        "batch_size": 128,
+        "batch_size": 64,
         "betas": [
             0.8,
             0.99

infer/modules/train/train.py

@@ -98,7 +98,6 @@ def train(exp_dir: str):
     hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
     hps.version = "v2"
     hps.gpus = "0"
-    hps.train.batch_size = 128
     hps.sample_rate = "40k"
     hps.if_f0 = 1
     hps.if_latest = 0