Update learning rate
Browse filesSigned-off-by: Aadhitya A <[email protected]>
app.py
CHANGED
@@ -559,7 +559,7 @@ def modelTFT(csv_file, prax):
|
|
559 |
hidden_size_range=(8, 64),
|
560 |
hidden_continuous_size_range=(8, 24),
|
561 |
attention_head_size_range=(1, 4),
|
562 |
-
learning_rate_range=(0.
|
563 |
dropout_range=(0.1, 0.3),
|
564 |
trainer_kwargs=dict(limit_train_batches=30),
|
565 |
reduce_on_plateau_patience=4,
|
@@ -791,7 +791,7 @@ def modelTFT_OpenGap(csv_file, prax):
|
|
791 |
hidden_size_range=(8, 64),
|
792 |
hidden_continuous_size_range=(8, 24),
|
793 |
attention_head_size_range=(1, 4),
|
794 |
-
learning_rate_range=(0.
|
795 |
dropout_range=(0.1, 0.3),
|
796 |
trainer_kwargs=dict(limit_train_batches=30),
|
797 |
reduce_on_plateau_patience=4,
|
|
|
559 |
hidden_size_range=(8, 64),
|
560 |
hidden_continuous_size_range=(8, 24),
|
561 |
attention_head_size_range=(1, 4),
|
562 |
+
learning_rate_range=(0.01, 0.1),
|
563 |
dropout_range=(0.1, 0.3),
|
564 |
trainer_kwargs=dict(limit_train_batches=30),
|
565 |
reduce_on_plateau_patience=4,
|
|
|
791 |
hidden_size_range=(8, 64),
|
792 |
hidden_continuous_size_range=(8, 24),
|
793 |
attention_head_size_range=(1, 4),
|
794 |
+
learning_rate_range=(0.01, 0.1),
|
795 |
dropout_range=(0.1, 0.3),
|
796 |
trainer_kwargs=dict(limit_train_batches=30),
|
797 |
reduce_on_plateau_patience=4,
|