提交 0584ef40 编写于 作者: R root

fix cuda number

上级 39aef4f6
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
export FLAGS_sync_nccl_allreduce=0 export FLAGS_sync_nccl_allreduce=0
export FLAGS_eager_delete_tensor_gb=1 export FLAGS_eager_delete_tensor_gb=1
export CUDA_VISIBLE_DEVICES=2 export CUDA_VISIBLE_DEVICES=0
if [[ ! -d pretrain_model/bert ]]; then if [[ ! -d pretrain_model/bert ]]; then
bash download_pretrain.sh bert bash download_pretrain.sh bert
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册