Internal change

PiperOrigin-RevId: 548230965
oxmah · Jul 14, 2023 · 5a0305c · 5a0305c
1 parent ff5e429
commit 5a0305c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/official/projects/yolo/configs/experiments/yolov7/detection/yolov7_gpu.yaml b/official/projects/yolo/configs/experiments/yolov7/detection/yolov7_gpu.yaml
@@ -69,6 +69,11 @@ task:
  use_sync_bn: true
  train_data:
  input_path: /readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*
+ shuffle_buffer_size: 5000
+ decoder:
+ simple_decoder:
+ regenerate_source_id: true
+ coco91_to_80: false
  is_training: true
  global_batch_size: 32
  dtype: float32
@@ -100,6 +105,10 @@ task:
  aug_scale_max: 1.9
  validation_data:
  input_path: /readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*
+ decoder:
+ simple_decoder:
+ regenerate_source_id: true
+ coco91_to_80: false
  is_training: false
  global_batch_size: 8
  dtype: float32