AdamP-Tensorflow
AdamP-Tensorflow copied to clipboard
TPU Error: EuclideanNorm OpKernel not compatible
Full error description raised on Tensorflow's TPU repo can be found here.
可以在此处找到 Tensorflow 的 TPU 存储库中提出的完整错误描述。
Perhaps, you can try replacing the "math_ops.euclidean_norm" with the following:
def euclidean_norm(x, axis, keepdims=False):
return math_ops.reduce_sum(x ** 2, axis=axis, keepdims=keepdims) ** 0.5