Update app.py

app.py

@@ -203,10 +203,10 @@ else:
             )
 if if_gpu_ok == True and len(gpu_infos) > 0:
     gpu_info = "\n".join(gpu_infos)
-    default_batch_size = min(mem) // 16
+    default_batch_size = min(mem) // 2
 else:
     gpu_info = i18n("很遗憾您这没有能用的显卡来支持您训练")
-    default_batch_size = 16
+    default_batch_size = 1
 gpus = "-".join([i[0] for i in gpu_infos])
 from lib.infer_pack.models import (
     SynthesizerTrnMs256NSFsid,
@@ -905,7 +905,7 @@ def train_index(exp_dir1, version19):
     # faiss.write_index(index, '%s/trained_IVF%s_Flat_FastScan_%s.index'%(exp_dir,n_ivf,version19))
     infos.append("adding")
     yield "\n".join(infos)
-    batch_size_add = 8192
+    batch_size_add = 16384
     for i in range(0, big_npy.shape[0], batch_size_add):
         index.add(big_npy[i : i + batch_size_add])
     faiss.write_index(
@@ -1145,7 +1145,7 @@ def train1key(
         % (model_log_dir, n_ivf, index_ivf.nprobe, exp_dir1, version19),
     )
     yield get_info_str("adding index")
-    batch_size_add = 8192
+    batch_size_add = 16384
     for i in range(0, big_npy.shape[0], batch_size_add):
         index.add(big_npy[i : i + batch_size_add])
     faiss.write_index(