Commit 999fd0f5 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TF Object Detection Team
Browse files

Changing default number of prefetch batches to -1 (i.e. Autotune).

PiperOrigin-RevId: 329318576
parent 3f8132e9
...@@ -25,9 +25,9 @@ enum InstanceMaskType { ...@@ -25,9 +25,9 @@ enum InstanceMaskType {
// Input type format: whether inputs are TfExamples or TfSequenceExamples. // Input type format: whether inputs are TfExamples or TfSequenceExamples.
enum InputType { enum InputType {
INPUT_DEFAULT = 0; // Default implementation, currently TF_EXAMPLE INPUT_DEFAULT = 0; // Default implementation, currently TF_EXAMPLE
TF_EXAMPLE = 1; // TfExample input TF_EXAMPLE = 1; // TfExample input
TF_SEQUENCE_EXAMPLE = 2; // TfSequenceExample Input TF_SEQUENCE_EXAMPLE = 2; // TfSequenceExample Input
} }
// Next id: 34 // Next id: 34
...@@ -72,7 +72,7 @@ message InputReader { ...@@ -72,7 +72,7 @@ message InputReader {
// to a small constant and increment linearly until the improvements become // to a small constant and increment linearly until the improvements become
// marginal or you exceed your cpu memory budget. Setting this to -1, // marginal or you exceed your cpu memory budget. Setting this to -1,
// automatically tunes this value for you. // automatically tunes this value for you.
optional int32 num_prefetch_batches = 20 [default = 2]; optional int32 num_prefetch_batches = 20 [default = -1];
// Maximum number of records to keep in reader queue. // Maximum number of records to keep in reader queue.
optional uint32 queue_capacity = 3 [default = 2000, deprecated = true]; optional uint32 queue_capacity = 3 [default = 2000, deprecated = true];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment