perf: set batch size to 128

app/train.py

@@ -79,7 +79,7 @@ def write_filelist(exp_dir: str) -> None:
         f.write("\n".join(opt))
 
 
-@zero(duration=300)
+@zero(duration=240)
 def train_model(exp_dir: str) -> str:
     shutil.copy("config.json", exp_dir)
     write_filelist(exp_dir)

config.json

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

infer/modules/train/train.py

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