diff --git a/training/opts.lua b/training/opts.lua index bad932e..41be173 100644 --- a/training/opts.lua +++ b/training/opts.lua @@ -34,7 +34,7 @@ function M.parse(arg) ------------- Training options -------------------- 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('-epochNumber', 1, 'Manual epoch number (useful on restarts)') -- GPU memory usage depends on peoplePerBatch and imagesPerPerson.