Spaces:
Running
Running
Add training parameters guide to the interface for better user understanding
Browse files
app.py
CHANGED
|
@@ -811,6 +811,25 @@ with gr.Blocks() as demo:
|
|
| 811 |
|
| 812 |
with gr.Row():
|
| 813 |
with gr.Column():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 814 |
epochs = gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Number of Epochs")
|
| 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")
|
|
@@ -862,4 +881,4 @@ if __name__ == "__main__":
|
|
| 862 |
if not load_vector_store():
|
| 863 |
print("Knowledge base not found. Please create it through the interface.")
|
| 864 |
|
| 865 |
-
demo.launch()
|
|
|
|
| 811 |
|
| 812 |
with gr.Row():
|
| 813 |
with gr.Column():
|
| 814 |
+
gr.Markdown("""
|
| 815 |
+
### Training Parameters Guide
|
| 816 |
+
|
| 817 |
+
**Number of Epochs**
|
| 818 |
+
- What it means: How many times the model will see all training data
|
| 819 |
+
- Higher = More learning but takes longer
|
| 820 |
+
- Recommended: 3 for small datasets, 1 for large datasets
|
| 821 |
+
|
| 822 |
+
**Batch Size**
|
| 823 |
+
- What it means: How many examples processed at once
|
| 824 |
+
- Higher = Faster training but needs more memory
|
| 825 |
+
- Recommended: 4 for 16GB RAM, 8 for 32GB RAM
|
| 826 |
+
|
| 827 |
+
**Learning Rate**
|
| 828 |
+
- What it means: How quickly the model learns
|
| 829 |
+
- Higher = Faster learning but may be less stable
|
| 830 |
+
- Recommended: 2e-4 (0.0002) for most cases
|
| 831 |
+
""")
|
| 832 |
+
|
| 833 |
epochs = gr.Slider(minimum=1, maximum=10, value=3, step=1, label="Number of Epochs")
|
| 834 |
batch_size = gr.Slider(minimum=1, maximum=32, value=4, step=1, label="Batch Size")
|
| 835 |
learning_rate = gr.Slider(minimum=1e-6, maximum=1e-3, value=2e-4, label="Learning Rate")
|
|
|
|
| 881 |
if not load_vector_store():
|
| 882 |
print("Knowledge base not found. Please create it through the interface.")
|
| 883 |
|
| 884 |
+
demo.launch()
|