Skip to content

Instantly share code, notes, and snippets.

@takada-at
Created January 21, 2021 04:23
Show Gist options
  • Save takada-at/bd43ad535b8643fd87304607bc9b2c7d to your computer and use it in GitHub Desktop.
Save takada-at/bd43ad535b8643fd87304607bc9b2c7d to your computer and use it in GitHub Desktop.
export IMAGE_URI=gcr.io/pj-kg-kpi/twitter_ai_platform
docker build -f Dockerfile -t $IMAGE_URI ./
docker push $IMAGE_URI
export MODEL_DIR=twitter_model_$(date +%Y%m%d_%H%M%S)
export REGION=us-central1
export JOB_NAME=twitter_search_training_$(date +%Y%m%d_%H%M%S)
gcloud ai-platform jobs submit training $JOB_NAME \
--region $REGION \
--master-image-uri $IMAGE_URI \
--scale-tier BASIC_GPU \
--python-version 3.7 \
-- \
--job-dir=gs://kg-kpi-ai-platform-test/$MODEL_DIR \
--epochs=1 --batch_size 8
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment