X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=dyncnn.git;a=blobdiff_plain;f=dyncnn.lua;h=839431ab40ccdd33d277c2927295ef40016c0ef2;hp=7cc531880ce18e680bd6a10b61b91a726bcd5ccd;hb=4cab0b04a02e270f4e6bce11a763ddfe0a2ad2ae;hpb=566fc0bed5110585f5e17fe3afdb60c30ecb2cba diff --git a/dyncnn.lua b/dyncnn.lua index 7cc5318..839431a 100755 --- a/dyncnn.lua +++ b/dyncnn.lua @@ -642,7 +642,7 @@ function trainModel(model, local startTime = sys.clock() for b = 1, validationData.nbSamples, batchSize do - fillBatch(trainData, b, batchSize, batch) + fillBatch(validationData, b, batchSize, batch) local output = model:forward(batch.input) accLoss = accLoss + l2criterion:forward(output, batch.target) nbBatches = nbBatches + 1