|
@@ -282,7 +282,7 @@ def build_ui():
|
|
|
)
|
|
)
|
|
|
max_samples = gr.Number(
|
|
max_samples = gr.Number(
|
|
|
label="Max Samples (0 = use all)",
|
|
label="Max Samples (0 = use all)",
|
|
|
- value=5000,
|
|
|
|
|
|
|
+ value=0,
|
|
|
info="Limit dataset size for faster experiments",
|
|
info="Limit dataset size for faster experiments",
|
|
|
)
|
|
)
|
|
|
|
|
|
|
@@ -299,7 +299,7 @@ def build_ui():
|
|
|
num_epochs = gr.Slider(1, 10, value=2, step=1, label="Epochs")
|
|
num_epochs = gr.Slider(1, 10, value=2, step=1, label="Epochs")
|
|
|
learning_rate = gr.Number(label="Learning Rate", value=2e-4)
|
|
learning_rate = gr.Number(label="Learning Rate", value=2e-4)
|
|
|
batch_size = gr.Slider(
|
|
batch_size = gr.Slider(
|
|
|
- 1, 8, value=1, step=1, label="Batch Size per Device"
|
|
|
|
|
|
|
+ 1, 32, value=16, step=1, label="Batch Size per Device"
|
|
|
)
|
|
)
|
|
|
grad_accum = gr.Slider(
|
|
grad_accum = gr.Slider(
|
|
|
1, 64, value=8, step=1, label="Gradient Accumulation Steps"
|
|
1, 64, value=8, step=1, label="Gradient Accumulation Steps"
|
|
@@ -373,7 +373,7 @@ def build_ui():
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
# Auto-refresh every 10 seconds using Timer
|
|
# Auto-refresh every 10 seconds using Timer
|
|
|
- timer = gr.Timer(value=10)
|
|
|
|
|
|
|
+ timer = gr.Timer(value=2)
|
|
|
timer.tick(
|
|
timer.tick(
|
|
|
fn=poll_status,
|
|
fn=poll_status,
|
|
|
outputs=[
|
|
outputs=[
|