Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
999fd0f5
Commit
999fd0f5
authored
Aug 31, 2020
by
A. Unique TensorFlower
Committed by
TF Object Detection Team
Aug 31, 2020
Browse files
Changing default number of prefetch batches to -1 (i.e. Autotune).
PiperOrigin-RevId: 329318576
parent
3f8132e9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
research/object_detection/protos/input_reader.proto
research/object_detection/protos/input_reader.proto
+4
-4
No files found.
research/object_detection/protos/input_reader.proto
View file @
999fd0f5
...
@@ -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
];
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment