From d388376101c7bbc6c8ddf15d172a113f48d56e11 Mon Sep 17 00:00:00 2001 From: qinduohao Date: Thu, 18 Apr 2019 13:51:39 +0800 Subject: [PATCH] readme add CUDA_VISIBLE_DEVICES description --- AutoDL Design/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/AutoDL Design/README.md b/AutoDL Design/README.md index 1b50e88..9173578 100644 --- a/AutoDL Design/README.md +++ b/AutoDL Design/README.md @@ -60,6 +60,8 @@ mid=460, average rewards=17.000 export FLAGS_fast_eager_deletion_mode=1 CUDA_VISIBLE_DEVICES=0 python -u main.py ``` +__注意:__ 这里训练需要使用两张卡来训练,Agent使用的卡为`CUDA_VISIBLE_DEVICES=0`(设置在启动`main.py`命令中);Trainer训练使用卡为`CUDA_VISIBLE_DEVICES=1`(设置在[autodl.py](https://github.com/PaddlePaddle/AutoDL/blob/master/AutoDL%20Design/autodl.py#L124)文件中) + 预期结果: 日志中 `average accuracy`逐步增大 -- GitLab