Fix train.py

mitmul · Aug 14, 2017 · 837c316 · 837c316
1 parent 5aa5956
commit 837c316
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chainercmd/train.py b/chainercmd/train.py
@@ -164,7 +164,7 @@ def train(args):
             evaluator = getattr(mod, values['name'])
             if evaluator is extensions.Evaluator:
                 evaluator = evaluator(
-                    valid_iter, model, device=devices['main'])
+                    valid_iter, model, device=args.gpus[0])
             else:
                 evaluator = evaluator(valid_iter, model.predictor)
             trainer.extend(