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()关闭梯度同时进行