Repository navigation
Expand file tree
/
Copy pathfetch-and-run.sh
More file actions
executable file
·88 lines (75 loc) · 2.87 KB
/
Copy pathfetch-and-run.sh
File metadata and controls
executable file
·88 lines (75 loc) · 2.87 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
#!/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_MODEL_TMPDIR: ${MINIGUI_MODEL_TMPDIR}"
echo "MINIGUI_BOARD_SIZE: ${MINIGUI_BOARD_SIZE}"
echo "MINIGUI_PORT: ${MINIGUI_PORT}"
echo "MINIGUI_HOST: ${MINIGUI_HOST}"
pyversion=$($MINIGUI_PYTHON --version)
echo "Python Version: ${pyversion}"
echo
echo "Downloading Model files:"
echo "--------------------------------------------------"
MODEL_SUFFIXES=( "data-00000-of-00001" "index" "meta" )
mkdir -p $MINIGUI_MODEL_TMPDIR
for suffix in "${MODEL_SUFFIXES[@]}"
do
file_to_check="${MINIGUI_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} $MINIGUI_MODEL_TMPDIR/
fi
done
# Assume models need to be frozen if .pb doesn't exist.
cd ..
model_path="${MINIGUI_MODEL_TMPDIR}/${MINIGUI_MODEL}"
if [[ ! -f "${model_path}.pb" ]]; then
echo
echo "Freezing model"
echo "--------------------------------------------------"
BOARD_SIZE=$MINIGUI_BOARD_SIZE $MINIGUI_PYTHON main.py freeze-graph \
$model_path --conv_width=128
fi
echo
echo "Running Minigui!"
echo "--------------------------------------------------"
echo "Model: $model_path"
echo "Size: $MINIGUI_BOARD_SIZE"
$MINIGUI_PYTHON minigui/serve.py \
--model="$model_path" \
--board_size="$MINIGUI_BOARD_SIZE" \
--port=$MINIGUI_PORT \
--host=$MINIGUI_HOST \
--python_for_engine=${MINIGUI_PYTHON} \
--engine=tf
}