Codex commited on
Commit
239a98e
·
1 Parent(s): 382222d

Increase batch size slider limits

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -884,14 +884,14 @@ def create_ui():
884
  with gr.Column(scale=1):
885
  gr.Markdown("### 4. SoVITS 训练")
886
  sovits_epochs = gr.Slider(1, 50, value=2, step=1, label="训练轮数")
887
- sovits_batch = gr.Slider(1, 4, value=1, step=1, label="批次大小")
888
  sovits_save_every = gr.Slider(1, 5, value=1, step=1, label="每隔多少轮导出")
889
  sovits_lr = gr.Slider(1e-5, 5e-4, value=1e-4, step=1e-5, label="学习率")
890
  sovits_btn = gr.Button("开始 SoVITS 训练", variant="primary", size="lg")
891
 
892
  gr.Markdown("### 5. GPT 训练")
893
  gpt_epochs = gr.Slider(1, 50, value=1, step=1, label="训练轮数")
894
- gpt_batch = gr.Slider(1, 4, value=1, step=1, label="批次大小")
895
  gpt_save_every = gr.Slider(1, 5, value=1, step=1, label="每隔多少轮导出")
896
  gpt_btn = gr.Button("开始 GPT 训练", variant="secondary")
897
 
 
884
  with gr.Column(scale=1):
885
  gr.Markdown("### 4. SoVITS 训练")
886
  sovits_epochs = gr.Slider(1, 50, value=2, step=1, label="训练轮数")
887
+ sovits_batch = gr.Slider(1, 8, value=1, step=1, label="批次大小")
888
  sovits_save_every = gr.Slider(1, 5, value=1, step=1, label="每隔多少轮导出")
889
  sovits_lr = gr.Slider(1e-5, 5e-4, value=1e-4, step=1e-5, label="学习率")
890
  sovits_btn = gr.Button("开始 SoVITS 训练", variant="primary", size="lg")
891
 
892
  gr.Markdown("### 5. GPT 训练")
893
  gpt_epochs = gr.Slider(1, 50, value=1, step=1, label="训练轮数")
894
+ gpt_batch = gr.Slider(1, 8, value=1, step=1, label="批次大小")
895
  gpt_save_every = gr.Slider(1, 5, value=1, step=1, label="每隔多少轮导出")
896
  gpt_btn = gr.Button("开始 GPT 训练", variant="secondary")
897