You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Jul 21, 2021. It is now read-only.
Firstly, thanks for sharing your pytorch version code of deformable convolution.
When I tried to port your code to another code which uses CUDA mode, it gives me an error saying like below.
@oeway Thanks for reply.
Actually in the deform_conv.py file there is a function called "th_batch_map_coordinates" which uses torch.range.
Anyway, you mean that there is no problem in usage of it and also did not the error occured in torch.range?
however, if you read the code, you will found that my torch.range is actually working in cpu mode, and after it may convert to cuda if the input is in cuda mode. So it still doesn't mean it caused by torch.range.
Come back to your question, yes, we have no problem when running in cuda mode.
Hello oeway
Firstly, thanks for sharing your pytorch version code of deformable convolution.
When I tried to port your code to another code which uses CUDA mode, it gives me an error saying like below.
'Type torch.cuda.FloatTensor doesn't implement stateless method range'
However, when I change it to CPU mode, there is no error.
It seems like 'torch.range' does not fit on GPU mode in torch. Then what should be the solution?
Thanks in advance
The text was updated successfully, but these errors were encountered: