Fixes for travis.

cmusatyalab · Jun 13, 2016 · f4567e8 · f4567e8
1 parent fea2fb4
commit f4567e8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/tests/openface_neural_net_training_tests.py b/tests/openface_neural_net_training_tests.py
@@ -64,7 +64,7 @@ def test_dnn_training():
            '-nEpochs', '10',
            '-epochSize', '1',
            '-cache', netWorkDir,
-           '-cuda', '-testing',
+           '-cuda', '-cudnn', '-testing',
            '-nDonkeys', '-1']
     p = Popen(cmd, stdout=PIPE, stderr=PIPE,
               cwd=os.path.join(openfaceDir, 'training'))

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

diff --git a/training/train.lua b/training/train.lua
@@ -96,10 +96,10 @@ function train()
       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
-         fixNans(mod.running_mean, string.format("%d-%s-%s", i, mod, 'running_mean'))
-         fixNans(mod.running_var, string.format("%d-%s-%s", i, mod, 'running_var'))
+         fixNans(mod.running_mean, string.format("%d-%s-%s", j, mod, 'running_mean'))
+         fixNans(mod.running_var, string.format("%d-%s-%s", j, mod, 'running_var'))
       end
    end