diff options
Diffstat (limited to 'src/caffe/proto/caffe.proto')
-rw-r--r-- | src/caffe/proto/caffe.proto | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto index 4be96963..8eb39b36 100644 --- a/src/caffe/proto/caffe.proto +++ b/src/caffe/proto/caffe.proto @@ -105,7 +105,9 @@ message SolverParameter { optional float stepsize = 12; // the stepsize for learning rate policy "step" optional string snapshot_prefix = 13; // The prefix for the snapshot. - + // whether to snapshot diff in the results or not. Snapshotting diff will help + // debugging but the final protocol buffer size will be much larger. + optional bool snapshot_diff = 14 [ default = false]; // Adagrad solver parameters // For Adagrad, we will first run normal sgd using the sgd parameters above // for adagrad_skip iterations, and then kick in the adagrad algorithm, with @@ -114,8 +116,8 @@ message SolverParameter { // the layer parameter specifications, as it will adjust the learning rate // of individual parameters in a data-dependent way. // WORK IN PROGRESS: not actually implemented yet. - optional float adagrad_gamma = 14; // adagrad learning rate multiplier - optional float adagrad_skip = 15; // the steps to skip before adagrad kicks in + optional float adagrad_gamma = 15; // adagrad learning rate multiplier + optional float adagrad_skip = 16; // the steps to skip before adagrad kicks in } // A message that stores the solver snapshots |