Iterations and hyper parameters should be created under init_scope as well

PiperOrigin-RevId: 239852113
This commit is contained in:
Sourabh Bajaj 2019-03-22 13:21:21 -07:00 committed by TensorFlower Gardener
parent 121cd87159
commit 1a02534e7d

View File

@ -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)