From 62f6fc9215f14161b98272ffa3f4d28ffd3a3d00 Mon Sep 17 00:00:00 2001 From: Brandon Amos Date: Tue, 10 Nov 2015 10:38:27 -0500 Subject: [PATCH] Add note to training new models. --- docs/training-new-models.md | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/docs/training-new-models.md b/docs/training-new-models.md index a9f344b..517de8a 100644 --- a/docs/training-new-models.md +++ b/docs/training-new-models.md @@ -1,6 +1,16 @@ -# Training new models -This repository also contains our training infrastructure to promote an +# Training new neural network models + +We have also released our training infrastructure to promote an open ecosystem and enable quicker bootstrapping for new research and development. + +The deep neural network (DNN) model produces the feature representation +of faces. +We provide pre-trained models trained with about 500k images that +can be used to train new face classifiers as shown in +[Demo 3](http://cmusatyalab.github.io/openface/demo-3-classifier/). +Training a new DNN model is for advanced users and should be done +with large datasets (>500k images) to improve the feature representation. + Warning: Training is computationally and memory expensive and takes a few weeks on our Tesla K40 GPU. Because of this, the training code assumes CUDA is installed.