问题是这样的,用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,当然可以,而且很简单,笔者这里的方法不一定最好,但很方便。
分三步走:
- 先下载模型:
在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∼