Error when running demo
rossjillian opened this issue · comments
When attempting to reproduce the demo, I get the following error:
RuntimeError: Expected tensor for argument #1 'input' to have the same type as tensor for argument #2 'rois'; but type torch.cuda.FloatTensor does not equal torch.cuda.DoubleTensor (while checking arguments for roi_pool_forward_kernel)
This might be related to: NVIDIA/apex#832. Unfortunately, I can't seem to find a way to convert input to float, as that code seems to lie within a torch module.
Just convert the datatype of the rois to that of input, and it will run!