提交 e24363d8 编写于 作者: R Rick Chao 提交者: TensorFlower Gardener

SidecarEvaluator: Export SidecarEvaluator as tf.keras.experimental.SidecarEvaluator.

PiperOrigin-RevId: 380865941
上级 88cabd5c
path: "tensorflow.keras.experimental.SidecarEvaluator"
tf_class {
is_instance: "<class \'keras.distribute.sidecar_evaluator.SidecarEvaluator\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
argspec: "args=[\'self\', \'model\', \'data\', \'checkpoint_dir\', \'steps\', \'max_evaluations\', \'callbacks\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\'], "
}
member_method {
name: "start"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
}
......@@ -20,6 +20,10 @@ tf_module {
name: "SequenceFeatures"
mtype: "<type \'type\'>"
}
member {
name: "SidecarEvaluator"
mtype: "<type \'type\'>"
}
member {
name: "WideDeepModel"
mtype: "<type \'type\'>"
......
......@@ -17,6 +17,7 @@
import tensorflow.compat.v2 as tf
from tensorflow.python.platform import tf_logging as logging
from tensorflow.python.util.tf_export import keras_export # pylint: disable=g-direct-tensorflow-import
_PRINT_EVAL_STEP_EVERY_SEC = 60.0
_ITERATIONS_UNINITIALIZED = -1
......@@ -42,6 +43,7 @@ def list_checkpoint_attributes(ckpt_dir_or_file):
return {name.split('/')[0] for name in variable_map.keys()}
@keras_export('keras.experimental.SidecarEvaluator', v1=[])
class SidecarEvaluator(object):
"""A class designed for a dedicated evaluator task.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册