提交 1232f05b 编写于 作者: K Katherine Wu 提交者: TensorFlower Gardener

Remove "oss_optimizer" identifier.

PiperOrigin-RevId: 381276605
上级 b9125b2c
......@@ -1468,21 +1468,3 @@ tf.__internal__.saved_model.load.register_revived_type(
min_consumer_version=1,
setter=RestoredOptimizer._set_hyper # pylint: disable=protected-access
)])
def create_optimizer_with_warning(unused_proto):
warnings.warn("IMPORTANT: Invalid Optimizer identifier detected, please "
"sync and re-save the model by June 24, 2021.")
return RestoredOptimizer()
# TODO(kathywu): Delete this registration.
tf.__internal__.saved_model.load.register_revived_type(
"oss_optimizer",
lambda _: False, # Prefer the "optimizer" identifier.
versions=[tf.__internal__.saved_model.load.VersionedTypeRegistration(
object_factory=create_optimizer_with_warning,
version=1,
min_producer_version=1,
min_consumer_version=1,
setter=RestoredOptimizer._set_hyper # pylint: disable=protected-access
)])
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册