Updated code for cuda 11 and torch 1.10+

[callzhang]

The code has been updated for newer versions of torch, cuda and other dependencies.

[Yanghojun]

@callzhang

Hi callzhang!

Thanks for your commit.

I want to operate your train code but I have one problem.

In your correlation.py code

ctx.save_for_backward(self, input1, input2)

return error in my desktop because save_for_backward function can save only Tensor. (self is Correlation class)

Did u solve this problem or Didn't you have any problem before?