Training: Pass device option to batch-represent.

cmusatyalab · Jun 20, 2016 · b82f6c3 · b82f6c3
1 parent eaf7a1e
commit b82f6c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/training/test.lua b/training/test.lua
@@ -44,7 +44,8 @@ function test()
    print(outDir)
    local cmd = batchRepresent
    if opt.cuda then
-      cmd = cmd .. ' -cuda '
+      assert(opt.device ~= nil)
+      cmd = cmd .. ' -cuda -device ' .. opt.device
    end
    cmd = cmd .. ' -batchSize ' .. opt.testBatchSize ..
       ' -model ' .. latestModelFile ..