search

escorciav / roi_pooling

ROIPooling for pytorch
51 stars 13 forks source link

Fix Dtype for torch.int #4

Closed escorciav closed 5 years ago

escorciav commented 5 years ago

For some reason, the previous version of pytorch&cupy didn't complain when pyinn.utils.Dtype return None as the data type for integer tensors.

We must amend these two lines during forward, and backward.

We should request pyinn to extend the Dtype function.