Iterations and hyper parameters should be created under init_scope as well
PiperOrigin-RevId: 239852113
This commit is contained in:
parent
121cd87159
commit
1a02534e7d
@ -421,9 +421,9 @@ class OptimizerV2(trackable.Trackable):
|
||||
var_list = [v for (_, v) in grads_and_vars]
|
||||
|
||||
# Create iteration if necessary.
|
||||
_ = self.iterations
|
||||
self._create_hypers()
|
||||
with ops.init_scope():
|
||||
_ = self.iterations
|
||||
self._create_hypers()
|
||||
self._create_slots(var_list)
|
||||
|
||||
self._prepare(var_list)
|
||||
|
Loading…
Reference in New Issue
Block a user