TypeError：无法将提供的值转换为 EagerTensor。提供的值：0.0 请求的 dtype：int64

Question

我正在尝试训练可从 tensorflow 官方模型中获得的变压器模型。我可以在 cpu 中进行训练而没有任何错误，但是当我尝试 gpu 时出现以下错误：

models/official/transformer/v2/transformer.py:143 call  *
    encoder_outputs = self.encode(inputs, attention_bias, training)

models/official/transformer/v2/transformer.py:166 encode
    embedded_inputs = self.embedding_softmax_layer(inputs)

TypeError: Cannot convert provided value to EagerTensor. Provided value: 0.0 Requested dtype: int64

我试过 tf.cast 但它似乎没有帮助。

Answer 1

我在使用另一个 Keras 函数时遇到了同样的错误。我的一个参数是错误的浮点数。