Update
This commit is contained in:
parent
f213b15014
commit
6b5d6303cc
@ -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():
|
||||||
|
Loading…
Reference in New Issue
Block a user