README_CN.md

生成requestment.txt相关依赖

pip install pipreqs
pipreqs . --encoding=utf8 --force

安装相关依赖

pip install -r requirements.txt

里面的torch版本如下安装

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

运行

python main.py --data=yelp

具体超参数请查看Params.py文件

超参数调优

1、写入hyperParam.yaml中，通过排列组合进行训练。(为了充分利用资源我们采用多线程异步训练)

运行

python HyperParamMain.py

2、在Hyper.py中写入调试命令顺序执行

运行

python Hyper.py

其他

日志文件生成在/Log中、模型对应日志文件名存在Saved中。