OrgLion-ML
/
openface
mirror of https://github.com/cmusatyalab/openface.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix luacheck warning.

This commit is contained in:
Brandon Amos 2016-02-06 19:24:38 -05:00
parent b43f510489
commit 0ec2de4572
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
util/print-network-table.lua
View File

@ -25,12 +25,10 @@ local net = createModel()
local img = torch.randn(1, 3, opt.imgDim, opt.imgDim) local img = torch.randn(1, 3, opt.imgDim, opt.imgDim)
net:forward(img) net:forward(img)
-- for i,module in ipairs(net:listModules()) do
for i=1,#net.modules do for i=1,#net.modules do
local module = net.modules[i] local module = net.modules[i]
local out = torch.typename(module) .. ": " local out = torch.typename(module) .. ": "
for j, sz in ipairs(torch.totable(module.output:size())) do for _, sz in ipairs(torch.totable(module.output:size())) do
-- print(sz)
out = out .. sz .. ', ' out = out .. sz .. ', '
end end
out = string.sub(out, 1, -3) out = string.sub(out, 1, -3)