From df4ed231e6827393e4f367ff44afcf81027925eb Mon Sep 17 00:00:00 2001 From: Zilize Date: Sun, 28 Mar 2021 20:22:00 +0800 Subject: [PATCH] FIX: Update README file --- README.md | 4 +++- README_CN.md | 4 +++- run_crslab.py | 2 +- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 7d6271c..1625c77 100644 --- a/README.md +++ b/README.md @@ -149,7 +149,7 @@ python run_crslab.py --config config/crs/kgsf/redial.yaml --save_data --save_sys In summary, there are following arguments in `run_crslab.py`: - `--config` or `-c`: relative path for configuration file(yaml). -- `--gpu` or `-g`: specify gpu id(s) to use, only support using a single gpu now. Defaults to cpu(-1). +- `--gpu` or `-g`: specify GPU id(s) to use, we now support multiple GPUs. Defaults to CPU(-1). - `--save_data` or `-sd`: save pre-processed dataset. - `--restore_data` or `-rd`: restore pre-processed dataset from file. - `--save_system` or `-ss`: save trained system. @@ -254,6 +254,8 @@ We welcome all contributions from bug fixes to new features and extensions. We expect all contributions discussed in the issue tracker and going through PRs. +We thank the nice contributions through PRs from [@shubaoyu](https://github.com/shubaoyu), [@ToheartZhang](https://github.com/ToheartZhang). + ## Citing diff --git a/README_CN.md b/README_CN.md index bdcb34d..1a60af5 100644 --- a/README_CN.md +++ b/README_CN.md @@ -148,7 +148,7 @@ python run_crslab.py --config config/crs/kgsf/redial.yaml --save_data --save_sys 总的来说，`run_crslab.py`有如下参数可供调用： - `--config` 或 `-c`：配置文件的相对路径，以指定运行的模型与数据集。 -- `--gpu` or `-g`：指定 gpu id，目前仅支持单 GPU，默认使用 CPU（-1）。 +- `--gpu` or `-g`：指定 GPU id，支持多 GPU，默认使用 CPU（-1）。 - `--save_data` 或 `-sd`：保存预处理的数据。 - `--restore_data` 或 `-rd`：从文件读取预处理的数据。 - `--save_system` 或 `-ss`：保存训练好的 CRS 系统。 @@ -259,6 +259,8 @@ python run_crslab.py --config config/crs/kgsf/redial.yaml --save_data --save_sys 如果想贡献代码，请先在 Issue 中提出问题，然后再提 PR。 +我们感谢 [@shubaoyu](https://github.com/shubaoyu), [@ToheartZhang](https://github.com/ToheartZhang) 通过 PR 为项目贡献的新特性。 + ## 引用 diff --git a/run_crslab.py b/run_crslab.py index ad8686d..b0f7f73 100644 --- a/run_crslab.py +++ b/run_crslab.py @@ -20,7 +20,7 @@ parser.add_argument('-c', '--config', type=str, default='config/crs/tgredial/tgredial.yaml', help='config file(yaml) path') parser.add_argument('-g', '--gpu', type=str, default='-1', - help='specify gpu id(s) to use, only support using a single gpu now. Defaults to cpu(-1).') + help='specify GPU id(s) to use, we now support multiple GPUs. Defaults to CPU(-1).') parser.add_argument('-sd', '--save_data', action='store_true', help='save processed dataset') parser.add_argument('-rd', '--restore_data', action='store_true',