train_label = tf.keras.backend.one_hot(train_label,3)
train_label = tf.one_hot(train_label,3)
give the following error in tensorflow 2.0.0
InternalError: Could not find valid device for node.
Node: {{node OneHot}}
train_label
should be a INT
type data. for example:
train_label = [1, 2 ,3]
train_label = tf.one_hot(train_label,3) // work
train_label = [1., 2., 3.]
train_label = tf.one_hot(train_label,3) // InternalError: Could not find valid device for node.
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.