search

zhanghang1989 / ResNeSt

ResNeSt: Split-Attention Networks
https://arxiv.org/abs/2004.08955
Apache License 2.0
3.22k stars 494 forks source link

Fixing Import Error #142

Closed D0miH closed 3 years ago

D0miH commented 3 years ago

Hi, as described in #141 loading the Models using PyTorch does not work anymore. This PR should fix the loading error when using PyTorch.

closes #141