Minor changes for the 2000 epoch learning.

diff --git a/README.txt b/README.txt
index e143b3c..1852663 100644 (file)
--- 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 (executable)
--- 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 (executable)
--- 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}"