Support of PyTorch 1.5 and 1.6

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.

https://hangzhang.org/PyTorch-Encoding/

MIT License

2.04k stars 452 forks source link

Closed zhihengli-UR closed 4 years ago

zhihengli-UR commented 4 years ago

I failed to compile https://github.com/zhanghang1989/PyTorch-Encoding/tree/master/encoding/lib with PyTorch 1.5 or 1.6. Is there any plan to support latest version of PyTorch?

zhanghang1989 commented 4 years ago

Sorry that the code does not work for pytorch 1.5 or 1.6. I don't have time to update the code recently.

I would recommend using the docker:

bash scripts/build_docker.sh
bash scripts/run_docker.sh

zhihengli-UR commented 4 years ago

Thank you for your reply!

zhanghang1989 commented 4 years ago

FYI, the new version should be compatible with pytorch 1.6 https://github.com/zhanghang1989/PyTorch-Encoding/pull/309