From c349b3c173838ad841c92602992f404783181b88 Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Tue, 26 Aug 2014 14:32:14 -0700 Subject: [PATCH] default ilsvrc solving to GPU --- examples/imagenet/alexnet_solver.prototxt | 1 + examples/imagenet/imagenet_solver.prototxt | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/imagenet/alexnet_solver.prototxt b/examples/imagenet/alexnet_solver.prototxt index 8581e99..a3aeae0 100644 --- a/examples/imagenet/alexnet_solver.prototxt +++ b/examples/imagenet/alexnet_solver.prototxt @@ -11,3 +11,4 @@ momentum: 0.9 weight_decay: 0.0005 snapshot: 10000 snapshot_prefix: "caffe_alexnet_train" +solver_mode: GPU diff --git a/examples/imagenet/imagenet_solver.prototxt b/examples/imagenet/imagenet_solver.prototxt index aedec41..1ad40fe 100644 --- a/examples/imagenet/imagenet_solver.prototxt +++ b/examples/imagenet/imagenet_solver.prototxt @@ -11,3 +11,4 @@ momentum: 0.9 weight_decay: 0.0005 snapshot: 10000 snapshot_prefix: "caffe_imagenet_train" +solver_mode: GPU -- 2.7.4