提交 3e8e2733 编写于 作者: S Sebastian Höffner 提交者: François Chollet

Using K.is_tensor and K.is_variable (#13307)

is_tensor_or_variable(x) is undefined and replaced by
K.is_tensor(x) or K.is_variable(x).

Fixes #13306.
上级 88ca804d
......@@ -134,7 +134,7 @@ class LossFunctionWrapper(Loss):
def get_config(self):
config = {}
for k, v in six.iteritems(self._fn_kwargs):
config[k] = K.eval(v) if is_tensor_or_variable(v) else v
config[k] = K.eval(v) if K.is_tensor(v) or K.is_variable(v) else v
base_config = super(LossFunctionWrapper, self).get_config()
return dict(list(base_config.items()) + list(config.items()))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册