恢复模型时出错(多个OpKernel注册与NodeDef匹配)

时间:2017-06-14 03:52:36

标签: tensorflow

尝试从检查点恢复模型时出现错误。

这是2017-06-13的每晚Windows GPU for python 3.5。

InvalidArgumentError (see above for traceback): 
Multiple OpKernel registrations match NodeDef 'Decoder/decoder/GatherTree = GatherTree[T=DT_INT32, _device="/device:CPU:0"](Decoder/decoder/TensorArrayStack_1/TensorArrayGatherV3, Decoder/decoder/TensorArrayStack
_2/TensorArrayGatherV3, Decoder/decoder/while/Exit_18)': 'op: "GatherTree" device_type: "GPU" constraint { name: "T" allowed_values { list { type: DT_INT32 } } }' and 'op: "GatherTree" device_type: "GPU" constraint { name: "T" allowed_values { list { type: DT_
INT32 } } }'[[Node: Decoder/decoder/GatherTree = GatherTree[T=DT_INT32, _device="/device:CPU:0"](Decoder/decoder/TensorArrayStack_1/TensorArrayGatherV3, Decoder/decoder/TensorArrayStack_2/TensorArrayGatherV3, Decoder/decoder/while/Exit_18)]]

该模型使用带有波束搜索的dynamic_decode,否则在不使用波束搜索进行解码时,在训练模式下工作正常。

关于这意味着什么或如何调试它的任何想法?

1 个答案:

答案 0 :(得分:0)

我前一天也遇到过同样的问题。结果是张量流中的bug。现在已经解决了,BeamSearchDecoder应该使用最新的tensorflow构建。