PK15946 comments

Repositories
Issues
Comments

Results 1 comments of


                                            PK15946

Trouble propagating a custom gradient through FlowNet2 model

Hi, I encounter the same issue too, and I fixed it by using [this](https://github.com/NVIDIA/flownet2-pytorch/blob/master/networks/channelnorm_package/functions/channelnorm.py), hope it can help you!