ierhon commited on
Commit
6109b5b
1 Parent(s): eb3116b

Set the maximum epoch amount to 128

Browse files
Files changed (1) hide show
  1. chatbot_constructor.py +1 -1
chatbot_constructor.py CHANGED
@@ -120,7 +120,7 @@ if __name__ == "__main__":
120
  gr.components.Slider(0, 0.01, value=0.0001, step=1e-8, label="Regularization L1"),
121
  gr.components.Slider(0, 0.5, value=0.1, step=1e-8, label="Dropout"),
122
  gr.components.Slider(1e-8, 0.01, value=0.001, step=1e-8, label="Learning rate"),
123
- gr.components.Slider(1, 64, value=128, step=1, label="Epochs"),
124
  gr.components.Slider(1, 256, value=88, step=1, label="Embedding size"),
125
  gr.components.Slider(1, 128, value=16, step=1, label="Input Length"),
126
  gr.components.Slider(1, 128, value=64, step=1, label="Convolution kernel count"),
 
120
  gr.components.Slider(0, 0.01, value=0.0001, step=1e-8, label="Regularization L1"),
121
  gr.components.Slider(0, 0.5, value=0.1, step=1e-8, label="Dropout"),
122
  gr.components.Slider(1e-8, 0.01, value=0.001, step=1e-8, label="Learning rate"),
123
+ gr.components.Slider(1, 128, value=16, step=1, label="Epochs"),
124
  gr.components.Slider(1, 256, value=88, step=1, label="Embedding size"),
125
  gr.components.Slider(1, 128, value=16, step=1, label="Input Length"),
126
  gr.components.Slider(1, 128, value=64, step=1, label="Convolution kernel count"),