"discover/amd_windows.go" did not exist on "8570c1c0ef73e89448f6724645f56b9b10efef44"
pipeline.proto 556 Bytes
Newer Older
Yinxiao Li's avatar
Yinxiao Li committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
syntax = "proto2";

package object_detection.protos;

import "third_party/tensorflow_models/object_detection/protos/pipeline.proto";

extend TrainEvalPipelineConfig {
  optional LstmModel lstm_model = 205743444;
}

// Message for extra fields needed for configuring LSTM model.
message LstmModel {
  // Unroll length for training LSTMs.
  optional int32 train_unroll_length = 1;

  // Unroll length for evaluating LSTMs.
  optional int32 eval_unroll_length = 2;

  // Depth of the lstm feature map.
  optional int32 lstm_state_depth = 3 [default = 256];
}