📝 Update README & Add requirements

cattidea · Jan 9, 2022 · cd2c914 · cd2c914
1 parent ce64f1f
commit cd2c914
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -92,8 +92,17 @@ bash ./download_dataset.sh
 
 ### step4: 训练
 
+```bash
+export PYTHONPATH=$PWD:$PYTHONPATH
+CUDA_VISIBLE_DEVICES='0, 1, 2, 3' python -m paddle.distributed.launch tools/train_retrieval.py --cfg_file configs/retrieval_train.yaml
+```
+
 ### step5: 测试
 
+```bash
+python tools/eval_retrieval.py --cfg_file configs/retrieval_test.yaml
+```
+
 ### 使用预训练模型进行预测
 
 ## 六、代码结构与详细说明
@@ -115,7 +124,7 @@ bash ./download_dataset.sh
 ├── tests                     # 测试文件
 ├── tools
 │   └── train_retrieval.py    # 训练脚本
-│   └── test_retrieval.py     # 测试脚本
+│   └── eval_retrieval.py     # 测试脚本
 └── requirement.txt           # 依赖包
 ```
 

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,4 @@
+paddlenlp>=2.2.1
+paddlepaddle-gpu>=2.2.1
+yacs==0.1.8
+boto3