From 5cbea5ca8a26719be70c974fab505e5b8695d9e4 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Thu, 24 Nov 2016 13:16:14 +0100 Subject: [PATCH] Minor changes for the 2000 epoch learning. --- README.txt | 4 ++-- dyncnn.lua | 2 +- run.sh | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.txt b/README.txt index e143b3c..1852663 100644 --- a/README.txt +++ b/README.txt @@ -19,7 +19,7 @@ It will (1) Generate the data-set of 40k triplets of images, (2) Train the deep network, and output validation results every 100 - epochs. This takes 15h on a GTX 1080 with cuda 8.0, cudnn 5.1, + epochs. This takes ~30h on a GTX 1080 with cuda 8.0, cudnn 5.1, and recent torch. (3) Generate two pictures of the internal activations. @@ -28,5 +28,5 @@ It will -- Francois Fleuret -Nov 6, 2016 +Nov 24, 2016 Martigny diff --git a/dyncnn.lua b/dyncnn.lua index c050344..6625579 100755 --- a/dyncnn.lua +++ b/dyncnn.lua @@ -97,7 +97,7 @@ cmd:option('-nbBlocks', 8) cmd:text('') cmd:text('Training') -cmd:option('-nbEpochs', 2000, 'nb of epochs for the heavy setting') +cmd:option('-nbEpochs', 1000, 'nb of epochs for the heavy setting') cmd:option('-learningRate', 0.1, 'learning rate') cmd:option('-batchSize', 128, 'size of the mini-batches') cmd:option('-nbTrainSamples', 32768) diff --git a/run.sh b/run.sh index 7ccfbb6..fe3a2e2 100755 --- a/run.sh +++ b/run.sh @@ -42,8 +42,8 @@ if [[ ! -d "${DYNCNN_DATA_DIR}" ]]; then --dir "${DYNCNN_DATA_DIR}" fi -# Train the model (takes 30h on a GTX 1080 with cuda 8.0, cudnn 5.1, -# and recent torch) +# Train the model (2000 epochs takes 30h on a GTX 1080 with cuda 8.0, +# cudnn 5.1, and recent torch) if [[ ! -f "${DYNCNN_RUNDIR}"/scheme_02000.t7 ]]; then ./dyncnn.lua -rundir "${DYNCNN_RUNDIR}" -- 2.39.5