Skip to content

Latest commit

 

History

History
49 lines (38 loc) · 1.46 KB

GCP.md

File metadata and controls

49 lines (38 loc) · 1.46 KB

GCP Setup

Provisioning

export PROJECT_ID=...
export BUCKET=...

gcloud config set project $PROJECT_ID
gcloud config set compute/zone us-central1-b
gcloud config set compute/region us-central1

gsutil mb -c standard -l us-central1 -b on gs://$BUCKET
ctpu up --name ml-1 --preemptible 

gcloud compute ssh ml-1

Training

export MODEL=lstm_v3 && \
export BUCKET=... && \
export TPU_NAME=ml-1 && \
export STORAGE_BUCKET=gs://$BUCKET && \
export MODEL_DIR=$STORAGE_BUCKET/$MODEL && \
export DATA_DIR=$STORAGE_BUCKET/data && \
export PYTHONPATH="$PYTHONPATH:/usr/share/models"

python3 $MODEL.py \
--model_dir=$MODEL_DIR \
--data_dir=$DATA_DIR \
--distribution_strategy=tpu \
--tpu=$TPU_NAME \
--train_epochs=100

Resources