Error when running demo

[rossjillian]

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.

[avinabsaha]

Just convert the datatype of the rois to that of input, and it will run!