run_ml_job.sh

gcloud ai-platform jobs submit training ${job_name} --job-dir ${job_dir} \
  --python-version 3.5   \
  --runtime-version 1.9  \
  --package-path ./trainer            `# trainer module `  \
  --module-name trainer.train         `# file train.py from trainer module`    \
  --region ${region}     \
  --scale-tier BASIC_GPU   `# single NVIDIA Tesla K80 GPU` \
  --  `# Next, the params for train.py` \
  --weights_stage "${job_dir}/weights_stage_exported_tiny.h5"   `# exported stage pre-trained model `  \
  --weights_final "${job_dir}/weights_final_exported_tiny.h5""  `# exported final pre-trained model `  \
  --anchors_file "gs://${bucket_name}/tiny_yolo_anchors.txt" \
  --annotation_file "gs://${bucket_name}/2012_train_tiny.txt" \
  --classes_file "gs://${bucket_name}/voc_classes_tiny.txt"