This commit is contained in:
bmaltais 2023-02-26 15:11:21 -05:00
parent f213b15014
commit 6b5d6303cc

View File

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