search

baowenbo / DAIN

Depth-Aware Video Frame Interpolation (CVPR 2019)
https://sites.google.com/view/wenbobao/dain
MIT License
8.19k stars 840 forks source link

Bug in backward pass of FlowProjectionLayer #141

Closed danier97 closed 11 months ago

danier97 commented 3 years ago

Hi, great work, well done!

I noticed bug in one of your custom packages, FlowProjectionLayer. Specifically, in line 43 of the file FlowProjectionLayer.py, tensors input1 and count are saved in the graph, but in the backward function on line 61, an additional variable output is included, which causes a ValueError.

Thanks.

laomao0 commented 2 years ago

I think that should be ``'input1, count = ctx.saved_tensors'.