Spaces:
Running
Running
Refactor training parameter guide for clarity and improve layout in the UI
Browse files
app.py
CHANGED
|
@@ -815,32 +815,24 @@ with gr.Blocks() as demo:
|
|
| 815 |
batch_size = gr.Slider(minimum=1, maximum=32, value=4, step=1, label="Batch Size")
|
| 816 |
learning_rate = gr.Slider(minimum=1e-6, maximum=1e-3, value=2e-4, label="Learning Rate")
|
| 817 |
|
|
|
|
|
|
|
|
|
|
| 818 |
gr.Markdown("""
|
| 819 |
<small>
|
| 820 |
-
|
| 821 |
-
|
| 822 |
-
|
| 823 |
-
|
| 824 |
-
-
|
| 825 |
-
|
| 826 |
-
|
| 827 |
-
|
| 828 |
-
|
| 829 |
-
- Higher = Faster but needs more RAM
|
| 830 |
-
- Lower = Slower but more stable
|
| 831 |
-
- 4 = Good for 16GB RAM
|
| 832 |
-
- 8 = Good for 32GB RAM
|
| 833 |
-
|
| 834 |
-
**Learning Rate:**
|
| 835 |
-
- Higher = Learns faster but may be unstable
|
| 836 |
-
- Lower = Learns slower but more reliable
|
| 837 |
-
- 2e-4 (0.0002) = Usually works best
|
| 838 |
-
- 1e-4 = Safer choice for fine-tuning
|
| 839 |
</small>
|
| 840 |
""")
|
| 841 |
|
| 842 |
-
|
| 843 |
-
training_output = gr.Textbox(label="Training Status", interactive=False)
|
| 844 |
|
| 845 |
with gr.Column():
|
| 846 |
analysis_btn = gr.Button("Generate Chat Analysis")
|
|
|
|
| 815 |
batch_size = gr.Slider(minimum=1, maximum=32, value=4, step=1, label="Batch Size")
|
| 816 |
learning_rate = gr.Slider(minimum=1e-6, maximum=1e-3, value=2e-4, label="Learning Rate")
|
| 817 |
|
| 818 |
+
train_btn = gr.Button("Start Training", variant="primary")
|
| 819 |
+
training_output = gr.Textbox(label="Training Status", interactive=False)
|
| 820 |
+
|
| 821 |
gr.Markdown("""
|
| 822 |
<small>
|
| 823 |
+
|
| 824 |
+
**Epochs:** (- Higher = Model learns more thoroughly) (- Best for small datasets: 3-5)
|
| 825 |
+
(- Lower = Faster training ) (- Best for large datasets: 1-2)
|
| 826 |
+
|
| 827 |
+
**Batch Size:** (- Higher = Faster but needs more RAM) (- 4 = Good for 16GB RAM)
|
| 828 |
+
(- Lower = Slower but more stable ) (- 8 = Good for 32GB RAM)
|
| 829 |
+
|
| 830 |
+
**Learning Rate:** (- Higher = Learns faster but may be unstable) (- 2e-4 (0.0002) = Usually works best )
|
| 831 |
+
(- Lower = Learns slower but more reliable ) (- 1e-4 = Safer choice for fine-tuning)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 832 |
</small>
|
| 833 |
""")
|
| 834 |
|
| 835 |
+
|
|
|
|
| 836 |
|
| 837 |
with gr.Column():
|
| 838 |
analysis_btn = gr.Button("Generate Chat Analysis")
|