From 07099654bd75eafa8a27bd4199b8bbdd5bf265ec Mon Sep 17 00:00:00 2001 From: rtz19970824 Date: Wed, 21 Mar 2018 16:11:17 +0800 Subject: [PATCH] a bash file for training --- AlphaGo/train.sh | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 AlphaGo/train.sh diff --git a/AlphaGo/train.sh b/AlphaGo/train.sh new file mode 100644 index 0000000..aecf1cd --- /dev/null +++ b/AlphaGo/train.sh @@ -0,0 +1,16 @@ +GPU_play=(0) +GPU_train=(3,4,5,6) +str_play='python play.py --data_path=./data/ --save_path=./go/ --game=go &' +str_train='python model.py &' +play_each_GPU=4 + +$str_train +echo 'Start training' +for gpu in $GPU +do +export CUDA_VISIBLE_DEVICES=$gpu +for ((i=1;i<=$play_each_GPU;i++)) +do +$str_play +done +done