Skip to content

sidkuma24/dist-cnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Commands for Training CNN:

gcloud config set compute/zone us-east1-c
gcloud config set project [YOUR_PROJECT_ID]
git clone https://github.com/sidkuma24/dist-cnn/
  • Create the template instance :
gcloud compute instances create template-instance \
--image-project ubuntu-os-cloud \
--image-family ubuntu-1604-lts \
--boot-disk-size 10GB \
--machine-type n1-standard-1

gcloud compute ssh template-instance

sudo apt-get update
sudo apt-get -y upgrade \
&& sudo apt-get install -y python-pip python-dev

sudo pip install tensorflow
  • Create Google Cloud Storage Bucket for MNIST
BUCKET="mnist-$RANDOM-$RANDOM"
gsutil mb -c regional -l us-east1 gs://${BUCKET}

sudo ./scripts/create_records.py
gsutil cp /tmp/data/train.tfrecords gs://${BUCKET}/data/
gsutil cp /tmp/data/test.tfrecords gs://${BUCKET}/data/```
  • Create template instance
gcloud compute instances set-disk-auto-delete template-instance \
--disk template-instance --no-auto-delete

gcloud compute instances delete template-instance

gcloud compute images create template-image \
--source-disk template-instance
  • Create Other instances:
gcloud compute instances create \
master-0 worker-0 worker-1 ps-0 \
--image template-image \
--machine-type n1-standard-2 \
--scopes=default,storage-rw
  • Train the model
./scripts/start-training.sh gs://${BUCKET}

About

Distributed CNN using TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published