OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Add note to training new models.

This commit is contained in:
Brandon Amos 2015-11-10 10:38:27 -05:00
parent 8ce5a137e0
commit 62f6fc9215
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
docs/training-new-models.md
View File

@ -1,6 +1,16 @@
# Training new models # Training new neural network models
This repository also contains our training infrastructure to promote an
We have also released our training infrastructure to promote an
open ecosystem and enable quicker bootstrapping for new research and development. 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 Warning: Training is computationally and memory expensive and takes a
few weeks on our Tesla K40 GPU. few weeks on our Tesla K40 GPU.
Because of this, the training code assumes CUDA is installed. Because of this, the training code assumes CUDA is installed.