don't know why this happened

akar43 / lsm

Code for Learnt Stereo Machines based on the NIPS 2017 paper

MIT License

251 stars 62 forks source link

don't know why this happened #3

Closed xbcbx closed 6 years ago

xbcbx commented 6 years ago

When I was going to train the VLSM, this errror happened, and I don't know why. Sincerely look for help.

zhaohui-yang commented 6 years ago

I meet this problem too. Have you solved this problem?

akar43 commented 6 years ago

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 6 years ago

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 6 years ago

Glad it helped. Closing this issue.