提交 b53d2518 编辑于 作者: Shanshan Wu's avatar Shanshan Wu 提交者: tensorflow-copybara
浏览文件

Add TODO on exploring `get_config` requirement for custom keras metrics.

PiperOrigin-RevId: 392931812
上级 594e012e
......@@ -226,6 +226,8 @@ def federated_aggregate_keras_metric(
# If type(metric) is subclass of another tf.keras.metric arguments passed
# to __init__ must include arguments expected by the superclass and
# specified in superclass get_config().
# TODO(b/197746608): finds a safer way of reconstructing the metric,
# default argument values in Metric constructors can cause problems here.
keras_metric = None
try:
# This is some trickery to reconstruct a metric object in the current
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册