about Correlation_forward_cuda

[lvzhengyi0204]

Hello：
I have configured the environment according to your requirements and executed the command
1：bash install_scripts.sh
but when I run the commant：python3 tools/video_inpaint.py --frame_dir ./demo/frames --MASK_ROOT ./demo/masks --img_size 512 832 --FlowNet2 --DFC --ResNet101 --Propagation
there has
a error: AttributeError: module 'models.FlowNet2_Models.correlation_package._ext.correlation' has no attribute 'Correlation_forward_cuda'

[nbei]

Have you checked the version of PyTorch? And here is another branch without installation of flownet2.

[lvzhengyi0204]

torch 0.4.0
torchvision 0.2.1
I completely configure the environment according to your requirements

[lvzhengyi0204]

the correlation has No this attribute:Correlation_forward_cuda

[lvzhengyi0204]

please help me~ hahaha

[nbei]

I suggest that you can just compile the correlation layer and then see whether any error information will be reported.

[lvzhengyi0204]

I gave up flownet2 and switched to lightflow, and I can experiment normally