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

Question

Я запускал в фастае модель 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!

Answer 1

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

pip install resnest --pre

Answer 2

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