experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Pass custom object to Keras learning rate scheduler deserialization logic.

Browse Source 
PiperOrigin-RevId: 247305938
This commit is contained in:
A. Unique TensorFlower 2019-05-08 15:36:01 -07:00 committed by TensorFlower Gardener
parent 6fb0fcc218
commit 8cfbcc53cf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/optimizer_v2/optimizer_v2.py
View File

@ -674,7 +674,7 @@ class OptimizerV2(trackable.Trackable):
if "learning_rate" in config: if "learning_rate" in config:
if isinstance(config["learning_rate"], dict): if isinstance(config["learning_rate"], dict):
config["learning_rate"] = learning_rate_schedule.deserialize( config["learning_rate"] = learning_rate_schedule.deserialize(
config["learning_rate"]) config["learning_rate"], custom_objects=custom_objects)
return cls(**config) return cls(**config)
def _serialize_hyperparameter(self, hyperparameter_name): def _serialize_hyperparameter(self, hyperparameter_name):