zhanghang1989 / ResNeSt

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

TypeError: load_state_dict_from_url() got an unexpected keyword argument 'check_hash' #105

Open YingLv1106 opened 4 years ago

YingLv1106 commented 4 years ago

File "/home/lvying/Pycharm_Object/segment/Segmentation_rgbd/toolbox/models/backbone/resnest.py", line 3, in model = torch.hub.load('zhanghang1989/ResNeSt', 'resnest50', pretrained=True) File "/home/lvying/anaconda3/lib/python3.6/site-packages/torch/hub.py", line 340, in load model = entry(*args, **kwargs) File "/home/lvying/.cache/torch/hub/zhanghang1989_ResNeSt_master/resnest/torch/resnest.py", line 40, in resnest50 resnest_model_urls['resnest50'], progress=True, check_hash=True)) TypeError: load_state_dict_from_url() got an unexpected keyword argument 'check_hash'

zhanghang1989 commented 4 years ago

I think your pytorch version is out-of-date. https://pytorch.org/docs/master/hub.html?highlight=load_state_dict_from_url#torch.hub.load_state_dict_from_url

You may update pytorch or remove that argument.

YingLv1106 commented 4 years ago

Hi: Thank you. When I remove the parameter 'check_hash=True', it's working.

------------------ 原始邮件 ------------------ 发件人: "zhanghang1989/ResNeSt" <notifications@github.com>; 发送时间: 2020年8月11日(星期二) 凌晨4:18 收件人: "zhanghang1989/ResNeSt"<ResNeSt@noreply.github.com>; 抄送: "吕营"<1012097251@qq.com>;"Author"<author@noreply.github.com>; 主题: Re: [zhanghang1989/ResNeSt] TypeError: load_state_dict_from_url() got an unexpected keyword argument 'check_hash' (#105)

I think your pytorch version is out-of-date. https://pytorch.org/docs/master/hub.html?highlight=load_state_dict_from_url#torch.hub.load_state_dict_from_url

You may update pytorch or remove that argument.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.