Open HandsLing opened 6 years ago
i have meet the same problem. can you solve it ?
Have you solved it? I met the same question with you
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/pool_5_grad/RoiPoolGrad}} = RoiPoolGrad[T=DT_FLOAT, pooled_height=7, pooled_width=7, spatial_scale=0.0625](conv5_3/Relu, roi-data/rois, pool_5:1, gradients/fc6/transpose_grad/transpose)]]
Have you solved it @tsstjj ? I have the same problem here.
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]