Pytorch-Correlation-extension
Pytorch-Correlation-extension copied to clipboard
Custom implementation of Corrleation Module
@ClementPinard Hi, thanks for your great work, I used it when training FlowNet2C. Install is well-done. I use ` SpatialCorrelationSampler(kernel_size=1, patch_size=21, stride=1, padding=0, dilation=1, dilation_patch=2) ` to replace the correlation...
how can this extension work on 3D images?
In the terminal, you can import, but you get a strange error in the code.  In the code: File "/home/**/**/**/**/**/**.py", line 282, in __init__ from spatial_correlation_sampler import SpatialCorrelationSampler File...
torch = 1.10 cuda = 11.3 gcc = 9.5.0 running install /home/anaconda3/envs/py38torch110/lib/python3.8/site-packages/setuptools/command/install.py:34: SetuptoolsDeprecationWarning: setup.py install is deprecated. Use build and pip and other standards-based tools. warnings.warn( /home/anaconda3/envs/py38torch110/lib/python3.8/site-packages/setuptools/command/easy_install.py:144: EasyInstallDeprecationWarning: easy_install command...
Hi, I am building a Dockerfile for an app which has as a dependency the Pytorch correlation extension. I get this error either building it manually or with pip. The...
I had cloned “https://github.com/ClementPinard/Pytorch-Correlation-extension” and used the commend: python setup.py install, but there are something worth. I don't know how to tackle that. Help me please. cuda version: 11.2 pytorch...
Hi, did anyone get this to work on AMD GPUs on Linux? I tried various things, including the hipconvonvert scripts, but there seems to be some mismatch between headers and...
I tried to implement v4.0.0 in the following environment: CUDA 11.7 on A100 Gpu Torch 1.12.1 in Linux. Installation using pip is successful but running in python failed. Errors: from...
It seems that this section it is not compilable fullgraph: ``` torch._dynamo.exc.Unsupported: call_method UserDefinedObjectVariable(forward) __call__ [TensorVariable(), TensorVariable(), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int)] {}...