Skip to content

Latest commit

 

History

History
 
 

resnet

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

Introduction

This sample shows how to run training jobs on Cloud Machine Learning Engine with Cloud TPUs using TensorFlow's tf.metrics.

This sample is adapted from the official samples for training ResNet-50 with Cloud TPUs to run on Cloud Machine Learning Engine.

Requirements

Steps

  1. Clone the repository.

    git clone https://github.com/GoogleCloudPlatform/cloudml-samples.git
    
  2. If you do not already have a Cloud Storage bucket, create one to be used for the training job.

    gsutil mb gs://[YOUR_GCS_BUCKET]
    export GCS_BUCKET="gs://[YOUR_GCS_BUCKET]"
    
  3. Run the sample. The included script will train ResNet-50 for 1024 steps using a fake dataset.

    cd cloudml-samples/tpu/training/resnet
    bash submit_resnet.sh