fix typos

nmcardoso · Aug 19, 2023 · fddec6f · fddec6f
1 parent efef510
commit fddec6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mergernet/estimators/base.py b/mergernet/estimators/base.py
@@ -263,8 +263,8 @@ def get_scheduler(self, scheduler: str, lr: float) -> tf.keras.optimizers.schedu
         initial_learning_rate=lr,
         decay_steps=self.hp.get('lr_decay_steps', default=40),
         alpha=self.hp.get('lr_decay_alpha', default=0.0),
-        warmup_target=self.get('lr_warmup_target', default=None),
-        warmup_steps=self.get('lr_warmup_steps', default=0)
+        warmup_target=self.hp.get('lr_warmup_target', default=None),
+        warmup_steps=self.hp.get('lr_warmup_steps', default=0)
       )
     elif scheduler == 'exponential':
       return tf.keras.optimizers.schedules.ExponentialDecay(