diff --git a/training/opts.lua b/training/opts.lua index d830535..e49025f 100644 --- a/training/opts.lua +++ b/training/opts.lua @@ -35,7 +35,7 @@ function M.parse(arg) ------------- Training options -------------------- 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)') -- GPU memory usage depends on peoplePerBatch and imagesPerPerson. cmd:option('-peoplePerBatch', 15, 'Number of people to sample in each mini-batch.')