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

Fixes for travis.

This commit is contained in:
Brandon Amos 2016-06-13 14:06:12 -04:00
parent fea2fb41d8
commit f4567e882e
3 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/openface_neural_net_training_tests.py
View File

@ -64,7 +64,7 @@ def test_dnn_training():
'-nEpochs', '10', '-nEpochs', '10',
'-epochSize', '1', '-epochSize', '1',
'-cache', netWorkDir, '-cache', netWorkDir,
'-cuda', '-testing', '-cuda', '-cudnn', '-testing',
'-nDonkeys', '-1'] '-nDonkeys', '-1']
p = Popen(cmd, stdout=PIPE, stderr=PIPE, p = Popen(cmd, stdout=PIPE, stderr=PIPE,
cwd=os.path.join(openfaceDir, 'training')) cwd=os.path.join(openfaceDir, 'training'))

2
training/opts.lua
View File

@ -1,5 +1,3 @@
local lfs = require 'lfs'
local M = { } local M = { }
-- http://stackoverflow.com/questions/6380820/get-containing-path-of-lua-file -- http://stackoverflow.com/questions/6380820/get-containing-path-of-lua-file

6
training/train.lua
View File

@ -96,10 +96,10 @@ function train()
end end
end end
for i, mod in ipairs(model:listModules()) do for j, mod in ipairs(model:listModules()) do
if torch.typename(mod) == 'nn.SpatialBatchNormalization' then if torch.typename(mod) == 'nn.SpatialBatchNormalization' then
fixNans(mod.running_mean, string.format("%d-%s-%s", i, mod, 'running_mean')) fixNans(mod.running_mean, string.format("%d-%s-%s", j, mod, 'running_mean'))
fixNans(mod.running_var, string.format("%d-%s-%s", i, mod, 'running_var')) fixNans(mod.running_var, string.format("%d-%s-%s", j, mod, 'running_var'))
end end
end end