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.
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.