模型运行出现的问题汇总和解决办法

1.服务器上面下载模型

        出现:An error happened while trying to locate the file on the Hub and we cannot find the requested files in the local cache. Please check your connection and try again or make sure your Internet connection is on.

        解决:HF_ENDPOINT=https://hf-mirror.com python xxx.py

2.pip install torch_sparse下载错误

        错误:ERROR: Could not build wheels for torch_sparse, which is required to install pyproject.toml-based projects

        解决:查看本机的pytorch版本,https://pytorch-geometric.com/whl/,官网里面选择匹配的版本

        复制上面的地址栏,pytorch-geometric.com/whl/torch-1.13.1+cu117.html

        

pip install torch_sparse==0.6.14 -f https://pytorch-geometric.com/whl/torch-1.13.1+cu117.html

3.model.eval()作用

切换到验证模式,主要影响模型中的特定层(如 Dropout 和 BatchNorm)的行为,一般配合torch.no_grad()关闭梯度同时进行

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值