Anyone know how to solve this bug?

danfeiX / scene-graph-TF-release

"Scene Graph Generation by Iterative Message Passing" code repository

http://cs.stanford.edu/~danfei/scene-graph/

MIT License

421 stars 130 forks source link

Anyone know how to solve this bug? #22

Open sim-l3 opened 6 years ago

sim-l3 commented 6 years ago

InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'RoiPoolGrad' with these attrs. Registered devices: [CPU], Registered kernels: device='GPU'; T in [DT_FLOAT]

 [[Node: gradients/rel_pool5_grad/RoiPoolGrad = RoiPoolGrad[T=DT_FLOAT, pooled_height=7, pooled_width=7, spatial_scale=0.0625](conv_out, Placeholder_2, rel_pool5:1, gradients/rel_fc6/Reshape_grad/Reshape)]]

Ubuntu:16.04 Tensorflow-gpu=0.12

raven0327 commented 3 years ago

did you solve this problem? i have encountered the same.