Skip to content
This repository was archived by the owner on Dec 17, 2021. It is now read-only.

Latest commit

 

History

History
39 lines (22 loc) · 1.26 KB

README.md

File metadata and controls

39 lines (22 loc) · 1.26 KB

Introduction

This sample shows how to run training jobs on AI Platform 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 AI Platform.

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