projects / platform / upstream / caffe.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: f6536b5)
default ilsvrc solving to GPU
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Tue, 26 Aug 2014 21:32:14 +0000 (14:32 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Tue, 26 Aug 2014 21:32:19 +0000 (14:32 -0700)
examples/imagenet/alexnet_solver.prototxt patch | blob | history
examples/imagenet/imagenet_solver.prototxt patch | blob | history

diff --git a/examples/imagenet/alexnet_solver.prototxt b/examples/imagenet/alexnet_solver.prototxt
index 8581e99..a3aeae0 100644 (file)
--- 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 (file)
--- 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
Domain: Machine Learning / ML Framework;