Skip to content

Commit

Permalink
[FIX](handlers) Fix optimizer initialization with bad keyword lr
Browse files Browse the repository at this point in the history
  • Loading branch information
wbenbihi committed Aug 25, 2022
1 parent c3460a5 commit ee76786
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion hourglass_tensorflow/handlers/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def init_handler(self, *args, **kwargs) -> None:
self._learning_rate = self._instantiate(self.config.learning_rate)
self._loss = self._instantiate(self.config.loss)
self._optimizer = self._instantiate(
self.config.optimizer, lr=self._learning_rate
self.config.optimizer, learning_rate=self._learning_rate
)
self._metrics = [obj.init() for obj in self.config.metrics]
self._callbacks = [obj.init() for obj in self.config.callbacks]
Expand Down

0 comments on commit ee76786

Please sign in to comment.