当我尝试使用adam
,sgd
或rmsprop
作为优化器来拟合模型时,它可以正常工作,但是adadelta
给出了这个奇怪错误:
InvalidArgumentError: Cannot assign a device for operation embedding_4/embeddings/Initializer/random_uniform/sub: Could not satisfy explicit device specification '' because the node {{colocation_node embedding_4/embeddings/Initializer/random_uniform/sub}} was colocated with a group of nodes that required incompatible device '/job:localhost/replica:0/task:0/device:GPU:0'. All available devices [/job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:GPU:0].
出什么问题了?