This commit is contained in:
bmaltais 2023-02-26 15:11:21 -05:00
parent f213b15014
commit 6b5d6303cc
1 changed files with 2 additions and 2 deletions

View File

@ -415,7 +415,7 @@ def gradio_training(
with gr.Row():
train_batch_size = gr.Slider(
minimum=1,
maximum=32,
maximum=64,
label='Train batch size',
value=1,
step=1,
@ -485,7 +485,7 @@ def gradio_training(
'SGDNesterov',
'SGDNesterov8bit'
],
value="AdamW",
value="AdamW8bit",
interactive=True,
)
with gr.Row():