Training: By default use an epoch size of 500.

This commit is contained in:
Brandon Amos 2016-01-08 10:58:21 -05:00
parent d7ae2a3885
commit 22cf225923
1 changed files with 1 additions and 1 deletions

View File

@ -34,7 +34,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', 1000, 'Number of batches per epoch') cmd:option('-epochSize', 500, 'Number of batches per epoch')
cmd:option('-testEpochSize', 0, 'Number of batches to test per epoch') cmd:option('-testEpochSize', 0, 'Number of batches to test 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.