FlowNet2C training and validation about pytorch-correlation-extension

[ClementPinard]

Hi, thanks for your interest in this repo.
Can you add some details ? What code do you use exactly ? Do you have a repo name ? I used this repo for FlowNetC in my own FlowNetPytorch repo without problem.

[Queenyy]

@ClementPinard Thanks for your quick reply. I use the NVIDIA/flownet2-pytorchhttps://github.com/NVIDIA/flownet2-pytorch repo. When i use its own correlation function to train FlowNet2C, the loss is bumping, but train FlowNet2S can get a right result. So I guess the bad result on FlowNet2C is because of correlation layer, thus i used your work to replace the original function. It seems that the loss is still bumping.
I use cuda 10.0, pytorch 1.2, gcc 4.4.7. is there some problems here?
Do you have other trick to train FlowNet2C?
Extremely Grateful.