【huggingface】【pytorch-image-models】导入模型报错:ConnectionResetError: [Errno 104] Connection reset by peer

在这里插入图片描述

问题是这样的,用https://github.com/huggingface/pytorch-image-models跑模型,很好用,里面支持很多模型,但是要用预训练模型,训练时用这个参数--pretrained,但如果没科学上网,就会报这个错误。

报错信息片段:

Loading pretrained weights from Hugging Face hub
ConnectionResetError: [Errno 104] Connection reset by peer

对,这其实就是要科学上网,但linux又没有,但是windows有,那可不可以先下载好模型,或者直接用下载好的模型,而不用代码中去下载?OK,当然可以,而且很简单,笔者这里的方法不一定最好,但很方便。

分三步走:

  1. 先下载模型:

https://huggingface.co/timm/直接搜索:

在这里插入图片描述
2. 把1中的文件下载到一个文件夹中
3. 按照下方方法修改代码

直接用修改pytorch-image-models\timm\models\_hub.py这里面
在这里插入图片描述
修改成:

cached_safe_file =  "/path/to/your-model/model.safetensors"

注意:一定要到下载模型的model.safetensors文件,其他文件也放在文件夹中。

到此就OK了,enjoy~

∼ O n e   p e r s o n   g o   f a s t e r ,   a   g r o u p   o f   p e o p l e   c a n   g o   f u r t h e r ∼ \sim_{One\ person\ go\ faster,\ a\ group\ of\ people\ can\ go\ further}\sim One person go faster, a group of people can go further

猜你喜欢

转载自blog.csdn.net/ThomasCai001/article/details/135286109