From 5cbea5ca8a26719be70c974fab505e5b8695d9e4 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
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