Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

TrainYourOwnYOLO: Training on AWS

If your local machine does not have a GPU, training could take a very long time. To speed things up use an AWS GPU instance.

Spinning up a GPU Instance

To spin up a GPU instance, go to EC2 and select Launch Instance. Then go to Deep Learning AMI (Ubuntu) Version xx.x and hit Select. Under instance type, select p2.xlarge as the Instance Type. Click through to Step 4 Add Storage and add at least 10 GB more than the default value. Proceed by hitting Review and Launch.

Deep_Learning_AMI

Starting the Training

Connect to your instance and follow the same steps as on your local machine. Make sure that all your Source Images are in TrainYourOwnYOLO/Data/Source_Images and that both

are up-to-date.