bostondiditeam/MV3D

run train.py error

Master658 opened this issue · 4 comments

i successfully run make.sh and data.py
when i run train.py, i meet this error
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'RoiPoolGrad' with these attrs. Registered devices: [CPU], Registered kernels:
device='GPU'; T in [DT_FLOAT]

anyone can help me ?thanks l lot

After reduce the version of tensorflow, it works.

@Master658
Hi, can you tell which version of tensorflow are you using?