提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
最近在学习图提示学习领域,所以读了这篇刚刚发布的HGPrompt,想尝试跑开源代码
一、配置环境
因为之前建错了 需要删除
删除conda虚拟环境 要删除GPPT这个环境
conda env remove -n GPPT
配置好环境后就看作者写的requirement来配置环境
只有这三个 按照正常pip install即可安装完
二、运行
1.第一次运行代码需要配置环境变量
一直报错 libcusparse.so.11: cannot open shared object file: No such file or dir
缺少文件 但是进入/home/wangtb/miniconda3/lib后能找到这个文件
如果缺少 可以来此链接下载链接 提取码: m2m2
如果仍然报错,则在 /home/your_name/.bashrc 文件最下面,加入下述一行。注意,此路径下需包含 libcusparse.so.11 文件。
export LD_LIBRARY_PATH="/home/your_name/anaconda3/lib/"
这是由于没配置环境变量引起的 在本地都知道下载torch后需要配置环境变量 但服务器不知道如何配置
首先切换到home目录
cd ~
用文本编辑器打开 .bashrc
vim .bashrc
最后在里面
最下面 一行 添加 上面的
export LD_LIBRARY_PATH="/home/your_name/anaconda3/lib/"
之后按esc 因为目前为insert插入状态 按esc后输入冒号 wq 即可保存文件 保存后环境变量配置好 不会再报错 最后在终端执行下面语句即可
source ~/.bashrc
2.运行代码
HGPrompt的代码中,有自建的utils中的py文件,需要调用 但是在服务器中调用并没有使用过,所以这里面我们还需要自行修改(在终端运行不需要)
如果是在pycharm中运行,需要对路径修改
这是原文件中
说明我们要读取文件下的methods/HPrompt/utils/pytorchtools,所以添加
这个步骤不是一次性完成,可以一次一次的上一层文件添加,直到能运行为止
最后就是对数据集的解压 因为这个代码是对ACM数据集的训练
所以我们将ACM在本地解压后通过XFTP上传到data中
先运行pretrain后进行100轮训练,训练完毕后生成以下参数
最后进行run.py 对模型进行test
总结
总体来说,HGprompt要求的包已经代码整体是很清爽很简单的。同时我之前因为偷懒不想在跑新代码时候更换虚拟环境,总是报错glibc版本不够,但是当我更换新的虚拟环境后就好了值得注意。