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

DNN Training: Halve default epochSize.

This commit is contained in:
Brandon Amos 2016-03-06 20:03:54 -05:00
parent f17a36c2fd
commit 4616f61ff8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/opts.lua
View File

@ -35,7 +35,7 @@ function M.parse(arg)
------------- Training options -------------------- ------------- Training options --------------------
cmd:option('-nEpochs', 1000, 'Number of total epochs to run') cmd:option('-nEpochs', 1000, 'Number of total epochs to run')
cmd:option('-epochSize', 500, 'Number of batches per epoch') cmd:option('-epochSize', 250, 'Number of batches per epoch')
cmd:option('-epochNumber', 1, 'Manual epoch number (useful on restarts)') cmd:option('-epochNumber', 1, 'Manual epoch number (useful on restarts)')
-- GPU memory usage depends on peoplePerBatch and imagesPerPerson. -- GPU memory usage depends on peoplePerBatch and imagesPerPerson.
cmd:option('-peoplePerBatch', 15, 'Number of people to sample in each mini-batch.') cmd:option('-peoplePerBatch', 15, 'Number of people to sample in each mini-batch.')