kryox64 commited on
Commit
b5f564b
1 Parent(s): dbae0ca

Update learning rate

Browse files

Signed-off-by: Aadhitya A <[email protected]>

Files changed (1) hide show
  1. app.py +2 -2
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.001, 0.1),
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.001, 0.1),
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,