Set the maximum epoch amount to 128
Browse files- 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,
|
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"),
|