hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

a bash file for training

Browse Source
This commit is contained in:
rtz19970824 2018-03-21 16:11:17 +08:00
parent f70dfb0559
commit 07099654bd
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
AlphaGo/train.sh Normal file
View File

@ -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