Repository navigation
Expand file tree
/
Copy pathfetch-and-run.sh
More file actions
executable file
·106 lines (92 loc) · 3.5 KB
/
Copy pathfetch-and-run.sh
File metadata and controls
executable file
·106 lines (92 loc) · 3.5 KB
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#!/bin/bash
# Copyright 2018 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
set -e
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
# Run in a sub-shell so we don't unexpectedly set new variables.
{
source ${SCRIPT_DIR}/minigui-common.sh
cd $SCRIPT_DIR
# TODO(Kashomon): Use common utils library instead of copy-pasta.
command -v gsutil >/dev/null 2>&1 || {
echo >&2 "gsutil command is not defined"
echo >&2 "Install Google Cloud SDK first:"
echo >&2 " https://cloud.google.com/sdk/downloads"
exit 1
}
echo "Using: the following options for Minigui launch:"
echo "--------------------------------------------------"
echo "MINIGUI_PYTHON: ${MINIGUI_PYTHON}"
echo "MINIGUI_BUCKET_NAME: ${MINIGUI_BUCKET_NAME}"
echo "MINIGUI_GCS_DIR: ${MINIGUI_GCS_DIR}"
echo "MINIGUI_MODEL: ${MINIGUI_MODEL}"
echo "MINIGUI_TMPDIR: ${MINIGUI_TMPDIR}"
echo "MINIGUI_BOARD_SIZE: ${MINIGUI_BOARD_SIZE}"
echo "MINIGUI_PORT: ${MINIGUI_PORT}"
echo "MINIGUI_HOST: ${MINIGUI_HOST}"
echo "MINIGUI_NUM_READS: ${MINIGUI_NUM_READS}"
pyversion=$($MINIGUI_PYTHON --version)
echo "Python Version: ${pyversion}"
echo
echo "Downloading Model files:"
echo "--------------------------------------------------"
model_tmpdir="${MINIGUI_TMPDIR}/models"
control_tmpdir="${MINIGUI_TMPDIR}/control"
mkdir -p ${model_tmpdir}
mkdir -p ${control_tmpdir}
MODEL_SUFFIXES=( "data-00000-of-00001" "index" "meta" )
for suffix in "${MODEL_SUFFIXES[@]}"
do
file_to_check="${model_tmpdir}/${MINIGUI_MODEL}.${suffix}"
echo "Checking for: ${file_to_check}"
if [[ ! -f "${file_to_check}" ]]; then
gsutil cp gs://${MINIGUI_BUCKET_NAME}/${MINIGUI_GCS_DIR}/${MINIGUI_MODEL}.${suffix} ${model_tmpdir}/
fi
done
# Assume models need to be frozen if .pb doesn't exist.
cd ..
model_path="${model_tmpdir}/${MINIGUI_MODEL}"
if [[ ! -f "${model_path}.pb" ]]; then
echo
echo "Freezing model"
echo "--------------------------------------------------"
BOARD_SIZE=$MINIGUI_BOARD_SIZE $MINIGUI_PYTHON freeze_graph.py \
--model_path=${model_path}
fi
echo
echo "Running Minigui!"
echo "--------------------------------------------------"
echo "Model: ${model_path}"
echo "Size: ${MINIGUI_BOARD_SIZE}"
control_path="${control_tmpdir}/${MINIGUI_MODEL}.ctl"
cat > ${control_path} << EOL
board_size=19
players = {
"${MINIGUI_MODEL}" : Player("${MINIGUI_PYTHON}"
" -u"
" gtp.py"
" --load_file=${model_path}"
" --minigui_mode=true"
" --num_readouts=${MINIGUI_NUM_READS}"
" --resign_threshold=-0.8"
" --verbose=2",
startup_gtp_commands=[],
environ={"BOARD_SIZE": str(board_size)}),
}
EOL
${MINIGUI_PYTHON} minigui/serve.py \
--control="${control_path}" \
--port "${MINIGUI_PORT}" \
--host "${MINIGUI_HOST}"
}