Kevin Fink
commited on
Commit
·
4dd1004
1
Parent(s):
e6614b8
dev
Browse files
app.py
CHANGED
@@ -140,7 +140,7 @@ try:
|
|
140 |
gr.Textbox(label="HF hub to push to after training"),
|
141 |
gr.Textbox(label="HF API token"),
|
142 |
gr.Slider(minimum=1, maximum=10, value=3, label="Number of Epochs", step=1),
|
143 |
-
gr.Slider(minimum=1, maximum=
|
144 |
gr.Slider(minimum=1, maximum=1000, value=1, label="Learning Rate (e-5)", step=1),
|
145 |
gr.Slider(minimum=1, maximum=100, value=1, label="Gradient accumulation", step=1),
|
146 |
],
|
|
|
140 |
gr.Textbox(label="HF hub to push to after training"),
|
141 |
gr.Textbox(label="HF API token"),
|
142 |
gr.Slider(minimum=1, maximum=10, value=3, label="Number of Epochs", step=1),
|
143 |
+
gr.Slider(minimum=1, maximum=500, value=1, label="Batch Size", step=1),
|
144 |
gr.Slider(minimum=1, maximum=1000, value=1, label="Learning Rate (e-5)", step=1),
|
145 |
gr.Slider(minimum=1, maximum=100, value=1, label="Gradient accumulation", step=1),
|
146 |
],
|