Rulga commited on
Commit
233a915
·
1 Parent(s): ee79eb5

Refactor training parameter guide for clarity and improve layout in the UI

Browse files
Files changed (1) hide show
  1. app.py +13 -21
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
- **Quick Parameter Guide:**
821
-
822
- **Epochs:**
823
- - Higher = Model learns more thoroughly
824
- - Lower = Faster training
825
- - Best for small datasets: 3-5
826
- - Best for large datasets: 1-2
827
-
828
- **Batch Size:**
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
- train_btn = gr.Button("Start Training", variant="primary")
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")