-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_commands.txt
47 lines (41 loc) · 1.26 KB
/
run_commands.txt
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
#
# Training in local
# ------------------------------------------------------------------------------------
TRAIN_DATA="~/Dataset/celeba/train.flist"
EVAL_DATA="~/Dataset/celeba/valid.flist"
OUTPUT_PATH="$(pwd)/ckpt"
RUNTIME_VERSION="2.0"
PYTHON_VERSION="3.7.4"
gcloud ai-platform local train \
--module-name trainer.task \
--package-path trainer/ \
--job-dir OUTPUT_PATH \
-- \
--train-files $TRAIN_DATA \
--eval-files $EVAL_DATA \
--train-steps 1000 \
--eval-steps 100
#
# Training in local
# ------------------------------------------------------------------------------------
TRAIN_DATA="gs://inpaint-dataset/celeba/train.flist"
EVAL_DATA="gs://inpaint-dataset/celeba/valid.flist"
OUTPUT_PATH="gs://inpaint-dataset/output_models/DFNet_celeba_tf2"
RUNTIME_VERSION="2.0"
PYTHON_VERSION="3.5"
REGION="asia-east1"
JOB_NAME="DFNet_tf2_v2_GPU"
gcloud ai-platform jobs submit training $JOB_NAME \
--module-name trainer.task \
--package-path trainer/ \
--job-dir $OUTPUT_PATH \
--region $REGION \
--runtime-version $RUNTIME_VERSION \
--python-version $PYTHON_VERSION \
--config config.yaml \
-- \
--train-files $TRAIN_DATA \
--eval-files $EVAL_DATA \
--train-steps 1000 \
--eval-steps 100 \
--verbosity DEBUG