2017-06-14 66 views
-1

嘗試從檢查點還原模型時出現錯誤。恢復模型時出錯(多個OpKernel註冊與NodeDef匹配)

這是2017-06-13上的python 3.5的夜間Windows GPU版本。

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波束搜索,而不是使用用於解碼波束搜索時在訓練模式,否則工作正常。

有關這意味着什麼或如何調試它的任何想法?

回答

0

我一天前也面臨同樣的問題。原來是張量流的bug。現在已經解決了,BeamSearchDecoder應該可以與tensorflow的最新版本一起工作。