index 939df91164c94371df279198dcec47e4da8201ca..e1be61c35e15780691c4b9ae34115857707fa82d 100644 (file)
repeated float diff = 6 [packed=true];
}
+// The BlobProtoVector is simply a way to pass multiple blobproto instances
+// around.
+message BlobProtoVector {
+ repeated BlobProto blobs = 1;
+}
+
message Datum {
optional int32 channels = 1;
optional int32 height = 2;
message NetParameter {
optional string name = 1; // consider giving the network a name
repeated LayerConnection layers = 2; // a bunch of layers.
- repeated string input = 3; // The input to the network
+ // The input blobs to the network.
+ repeated string input = 3;
+ // The dim of the input blobs. For each input blob there should be four
+ // values specifying the num, channels, height and width of the input blob.
+ // Thus, there should be a total of (4 * #input) numbers.
+ repeated int32 input_dim = 4;
+ // Whether the network will force every layer to carry out backward operation.
+ // If set False, then whether to carry out backward is determined
+ // automatically according to the net structure and learning rates.
+ optional bool force_backward = 5 [ default = false ];
}
message SolverParameter {
// 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 = 16 [ default = false];
-
- // the asynchronous solver stuff
- optional string tcp_port = 20 [ default = "20901"];
- optional string tcp_server = 21;
- optional int32 communication_interval = 22;
+ // the mode solver will use: 0 for CPU and 1 for GPU. Use GPU in default.
+ optional int32 solver_mode = 17 [default = 1];
}
// A message that stores the solver snapshots