search

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 450 forks source link

Support Pytorch 1.7 or above #380

Open qiulesun opened 3 years ago

qiulesun commented 3 years ago

Does the repo support pytorch1.7 or above ? If not, could you consider updating the repo to support that for the use of 3090 GPUs ?

zhanghang1989 commented 3 years ago

I am not actively maintaining this repo. The ImageNet training code has been migrated to https://github.com/zhanghang1989/ResNeSt