Skip to content
This repository was archived by the owner on Mar 11, 2026. It is now read-only.

Commit 69b78e6

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

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tensorflow_addons/optimizers/lookahead.py

Lines changed: 1 addition & 0 deletions
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)