don't know why this happened
xbcbx opened this issue · 4 comments
xbcbx commented
zhaohui-yang commented
I meet this problem too. Have you solved this problem?
akar43 commented
I remember getting this error when operating in CPU mode. For some reason the gather_nd
op in Tensorflow does not behave the same way on CPU and GPU (at least did not when I last tested it). The code should work completely fine if you have a GPU though.
zhaohui-yang commented
I have solved this problem as @akar43 said. I specified the GPU mode and all worked fine. Thanks a lot~ This should be some mismatch in tersorflow implementation between CPU and GPU.
akar43 commented
Glad it helped. Closing this issue.