ChessAI/run_train.sh at master · MaxDillon/ChessAI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!/bin/bash

lead=0
skip=0
fast=0
dest=/dev/null
device=0

for arg in "$@"; do
    case $arg in
        -lead)  lead=1; shift;;
        -skip)  skip=1; shift;;
        -fast)  fast=1; shift;;
        -show)  dest=/dev/stdout; shift;;
        -dev1)  device=1; shift;;
    esac
done

while true; do
    if [[ $skip == 0 ]]; then
        if [[ $fast == 0 ]]; then
            ./mi_play.sh chess2 \
                -n 10 \
                -white "model2:model.chess2.Model015c.*" \
                -wargs iter=400,expl=0.15,temp=0.3,pexp 2.0,unif=1.0 \
                -black "model2:model.chess2.Model015c.*" \
                -bargs iter=400,expl=0.15,temp=0.3,pexp 2.0,unif=1.0 \
		-device $device \
                > $dest
        else
            ./mi_play.sh chess2 \
                -n 25 \
                -white "model2:model.chess2.Model015c.*" \
                -wargs iter=400,expl=0.15,temp=0.3,pexp 2.0,unif=1.0 \
                -black "model2:model.chess2.Model015c.*" \
                -bargs iter=400,expl=0.15,temp=0.3,pexp 2.0,unif=1.0 \
                -one false \
                -fast true \
                -mindepth 30 \
                -rollback 6 \
		-device $device \
                > $dest
        fi
    fi

    if [[ $lead != 0 ]]; then
        ./fetch.sh
        ./reshuffle.sh
        ./mi_train.sh chess2 \
            -data shuffled.chess2 \
            -from $(ls -t model.chess2.Model015c.* | head -n 1) \
            -valuemult 0.95 \
            -metf 0.7 \
            -lastn 150 \
            -batch 350 \
            -drawweight 0.2 \
            -saveevery 1000 \
            -updates 1 \
            -rate 5e-3
    fi

    if [ -e exit_run_train ]; then
	exit
    fi
done