init

app.py

@@ -58,7 +58,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             output_dir='./results',
             eval_strategy="epoch",
             save_strategy='epoch', 
-            learning_rate=lr*0.00001,
+            learning_rate=lr*0.000001,
             per_device_train_batch_size=int(batch_size),
             per_device_eval_batch_size=1, 
             num_train_epochs=int(num_epochs),
@@ -109,8 +109,8 @@ try:
             gr.Textbox(label="HF hub to push to after training"),
             gr.Textbox(label="HF API token"),
             gr.Slider(minimum=1, maximum=10, value=3, label="Number of Epochs", step=1),
-            gr.Slider(minimum=1, maximum=16, value=4, label="Batch Size", step=1),
-            gr.Slider(minimum=1, maximum=1000, value=50, label="Learning Rate (e-5)", step=1),
+            gr.Slider(minimum=1, maximum=16, value=1, label="Batch Size", step=1),
+            gr.Slider(minimum=1, maximum=1000, value=1, label="Learning Rate (e-5)", step=1),
             gr.Slider(minimum=1, maximum=100, value=1, label="Gradient accumulation (e-1)", step=1), 
         ],
         outputs="text",