KumapowerLIU / CSA-inpainting

Coherent Semantic Attention for image inpainting(ICCV 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A suggestion for more acceralted version of code

Zhaoyi-Yan opened this issue · comments

I find that your code is mainly based on deprecated code in pytorch 0.3 branch https://github.com/Zhaoyi-Yan/Shift-Net_pytorch/tree/pytorch0.3.1

However, this branch is slow, now the code in the master branch is must faster and is fully-parallel https://github.com/Zhaoyi-Yan/Shift-Net_pytorch/blob/master/models/shift_net/InnerShiftTripleFunction.py#L36-L49

You may upgrade your code to make it run much faster.

commented

Thank you. There are too many deadlines recently。 I will upgrade my code soon and do a demo. Your work is very good and it has inspired me a lot.