Skip to content

Commit 69b78e6

Browse files
authored
Lookahead tracking base_optimizer
Inital fix of #2094 #2102
1 parent 6329e1a commit 69b78e6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tensorflow_addons/optimizers/lookahead.py

+1
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ def __init__(
8080
self._set_hyper("sync_period", sync_period)
8181
self._set_hyper("slow_step_size", slow_step_size)
8282
self._initialized = False
83+
self._track_trackable(optimizer, 'base_optimizer')
8384

8485
def _create_slots(self, var_list):
8586
self._optimizer._create_slots(

0 commit comments

Comments
 (0)