From 0d51c40f6b70c41fa2f04dbf7528b3ed62bfb4b4 Mon Sep 17 00:00:00 2001
From: nagadomi <nagadomi@nurs.or.jp>
Date: Tue, 16 Oct 2018 18:17:47 +0000
Subject: [PATCH] lbp 512->128

---
 train.lua | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/train.lua b/train.lua
index 8b051ec..c95f4f0 100644
--- a/train.lua
+++ b/train.lua
@@ -390,15 +390,15 @@ local function create_criterion(model)
       return aux:cuda()
    elseif settings.loss == "lbp" then
       if reconstruct.is_rgb(model) then
-	 return w2nn.RandomBinaryCriterion(3, 512):cuda()
+	 return w2nn.RandomBinaryCriterion(3, 128):cuda()
       else
-	 return w2nn.RandomBinaryCriterion(1, 512):cuda()
+	 return w2nn.RandomBinaryCriterion(1, 128):cuda()
       end
    elseif settings.loss == "aux_lbp" then
       if reconstruct.is_rgb(model) then
-	 return w2nn.AuxiliaryLossCriterion(w2nn.RandomBinaryCriterion, {3, 512}):cuda()
+	 return w2nn.AuxiliaryLossCriterion(w2nn.RandomBinaryCriterion, {3, 128}):cuda()
       else
-	 return w2nn.AuxiliaryLossCriterion(w2nn.RandomBinaryCriterion, {1, 512}):cuda()
+	 return w2nn.AuxiliaryLossCriterion(w2nn.RandomBinaryCriterion, {1, 128}):cuda()
       end
    else
       error("unsupported loss .." .. settings.loss)