Training: By default use an epoch size of 500.
This commit is contained in:
parent
d7ae2a3885
commit
22cf225923
|
@ -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.
|
||||||
|
|
Loading…
Reference in New Issue