Rulga commited on
Commit
79a395a
·
1 Parent(s): a442043

Add training parameters guide to the interface for better user understanding

Browse files
Files changed (1) hide show
  1. app.py +20 -1
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()