Проблема ResNeSt: load_state_dict_from_url () получил неожиданный аргумент ключевого слова 'check_ha sh' - PullRequest
0 голосов
/ 20 июня 2020

Я запускал в фастае модель ResNeSt, но возникла проблема.

learn = fastai.vision.cnn_learner(data, resnest50(pretrained=True), metrics=[accuracy])

А вот ошибка:

TypeError                                 Traceback (most recent call last)
<ipython-input-4-cff6dd69b503> in <module>
----> 1 learn = fastai.vision.cnn_learner(data, resnest50(pretrained=True), metrics=[accuracy])

F:\Program Files\Python\lib\site-packages\resnest\torch\resnest.py in resnest50(pretrained, root, **kwargs)
     38     if pretrained:
     39         model.load_state_dict(torch.hub.load_state_dict_from_url(
---> 40             resnest_model_urls['resnest50'], progress=True, check_hash=True))
     41     return model
     42 

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

Thanks for your help!

Ответы [ 2 ]

1 голос
/ 09 июля 2020

Недавно мы обновили URL-адрес модели, установите последнюю версию ResNeSt https://github.com/zhanghang1989/ResNeSt

pip install resnest --pre
0 голосов
/ 20 июня 2020

Ну, я проверил свой исходный код pytorch и обнаружил, что код load_state_dict_from_url() в моем P C отличается от официального кода. Думаю, это проблема версии. Я не знаю, нужно ли мне обновить версию pytorch. Если мой cuda не будет совместим с новым фонариком?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...