Skip to content

Instantly share code, notes, and snippets.

@crawles
Last active May 22, 2018 19:20
Show Gist options
  • Save crawles/c90223200389fa705648e2de083c6950 to your computer and use it in GitHub Desktop.
Save crawles/c90223200389fa705648e2de083c6950 to your computer and use it in GitHub Desktop.
TRAIN_DATA_PATHS=path/to/training/data
OUTPUT_DIR=path/to/output/location
JOBNAME=my_ml_job_$(date -u +%y%m%d_%H%M%S)
REGION='us-central1'
BUCKET='my-bucket'
gcloud ml-engine jobs submit training $JOBNAME \
--package-path=$PWD/my_model_package/trainer \
--module-name=trainer.task \
--region=$REGION \
--staging-bucket=gs://$BUCKET \
--scale-tier=BASIC \
--runtime-version=1.8 \
-- \
--train_data_paths $TRAIN_DATA_PATHS \
--output_dir $OUTPUT_DIR\
--batch_size 100\
--hidden_units 50 25 10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment