Politrees commited on
Commit
c6b8c09
1 Parent(s): 8fb3190

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -203,10 +203,10 @@ else:
203
  )
204
  if if_gpu_ok == True and len(gpu_infos) > 0:
205
  gpu_info = "\n".join(gpu_infos)
206
- default_batch_size = min(mem) // 2
207
  else:
208
  gpu_info = i18n("很遗憾您这没有能用的显卡来支持您训练")
209
- default_batch_size = 1
210
  gpus = "-".join([i[0] for i in gpu_infos])
211
  from lib.infer_pack.models import (
212
  SynthesizerTrnMs256NSFsid,
 
203
  )
204
  if if_gpu_ok == True and len(gpu_infos) > 0:
205
  gpu_info = "\n".join(gpu_infos)
206
+ default_batch_size = min(mem) // 16
207
  else:
208
  gpu_info = i18n("很遗憾您这没有能用的显卡来支持您训练")
209
+ default_batch_size = 16
210
  gpus = "-".join([i[0] for i in gpu_infos])
211
  from lib.infer_pack.models import (
212
  SynthesizerTrnMs256NSFsid,