BasicSR
BasicSR copied to clipboard
Error in testing EDVR
error in modulated_deformable_im2col_cuda:invalid device function
I met same problem with you,have you solved it?
@automatic0313 It may be because the implementation of dcn.
Try the latest codes, which use the official torchvision.ops.deform_conv2d (I have verified that they have the same forward results)
(Note that the torchvision version should be >=0.9.0, therefore pytorch version >=1.8.0)
https://github.com/xinntao/BasicSR/blob/1ca522852049f48504afaab9df588735abbcec9b/basicsr/archs/arch_util.py#L230-L235