没办法编译corrlation模块

[ChaoyiXie]

我是Windows平台下cuda10.2 torch1.9.0，没办法编译你提供的代码种的corrlation模块，然后用你说的那个第三方的库https://github.com/ClementPinard/Pytorch-Correlation-extension 编译过了，但是怎么替换你网络中的self.corr；主要是这两个接口参数完全不一样，spatial_correlation_sample( kernel_size=3,patch_size=1, stride=2,padding=0,dilation=2,dilation_patch=1)；你的接口是 Correlation(pad_size=4, kernel_size=1, max_displacement=4, stride1=1, stride2=1, corr_multiply=1)；怎么一一对应起来

[ltkong218]

Maybe your CUDA and PyTorch version are relative high. CUDA 10.0 with PyTorch 1.2.0 are OK.

[hin7141]

@ChaoyiXie did you manage to solve this?