a bash file for training

2018-03-21 16:11:17 +08:00 · 2018-03-21 16:11:17 +08:00 · 07099654bd
commit 07099654bd
parent f70dfb0559
1 changed files with 16 additions and 0 deletions
--- a/AlphaGo/train.sh
+++ 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