CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

lbp 512->128

Browse source
This commit is contained in:
nagadomi 2018-10-16 18:17:47 +00:00
parent 4b9196deaa
commit 0d51c40f6b
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
train.lua
View file

@ -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)