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
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