search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
620 stars 123 forks source link

I already did what @fabiopk said but I am still getting this error #61

Closed Egyonic closed 12 months ago

Egyonic commented 1 year ago 
          I already did what @fabiopk said but I am still getting this error

AttributeError: module 'correlation' has no attribute 'FunctionCorrelation'

Originally posted by @RokiaAbdeen in https://github.com/sniklaus/pytorch-pwc/issues/19#issuecomment-698073223

I also got this problem. My env is : torch 1.12.1+cu113 torchvision 0.13.1+cu113 cupy-cuda113 10.6.0

File "/work/pytorch-liteflownet/compute_flow_vimeo.py", line 179, in forward tenCorrelation = torch.nn.functional.leaky_relu(input=correlation.FunctionCorrelation(tenFirst=tenFeaturesFirst, tenSecond=tenFeaturesSecond, intStride=1), negative_slope=0.1, inplace=False) AttributeError: module 'correlation' has no attribute 'FunctionCorrelation

anyone has the same problem?

sniklaus commented 1 year ago

What does python -c "print(__import__('torch').version.cuda)" print?

And what about python -c "__import__('cupy').show_config()"?

sniklaus commented 12 months ago

Closing due to inactivity, feel free to reopen should this still be an issue.