Fixes for travis.

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

training/opts.lua

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

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