Skip to content
Deep Learning Models optimized for AWS
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
.github Creating initial file from template Jul 19, 2018
hpc-cluster remove duplicated image Jul 20, 2018
models/resnet update log of evaluation Mar 19, 2019
utils Updating scripts to work better on DLAMI and train on 256GPUs in 14mi… Jan 4, 2019 Creating initial file from template Jul 19, 2018 Creating initial file from template Jul 19, 2018

Deep Learning Models

This repository contains scripts to train deep learning models optimized to run well on AWS. Apart from scripts to build and train the model, we also share here scripts to setup a high performing cluster for deep learning using AWS, and preprocessing scripts to prepare datasets.

Currently, it has scripts for training Resnet50 with Imagenet using Apache MXNet and Tensorflow. Feel free to create a Github issue here if you have any questions.

Manual cluster setup

  • Ensure that the security group of the instances allows connections through any port from within the same security group.
  • Ensure that your instances in the cluster have passwordless ssh set up. You should be able to do ssh IP1 where IP1 can be the IP of any node in the cluster, from any node in the cluster. One easy way to do this would be to use Agent Forwarding. Here is how to enable that.
eval `ssh-agent`
ssh-add key.pem

Running on AWS EC2 Deep Learning AMI

Make sure to use the script which handles the docker interface and conda environments.

Production ready cluster setup

Check out hpc-cluster in the repository which sets up a high performance cluster for deep learning. It uses best practices such as bastion hosts and a BeeGFS distributed file system across all nodes in the cluster for high performant store for the dataset.

License Summary

This sample code is made available under a modified MIT license. See the LICENSE file.

You can’t perform that action at this time.