Skip to content

GitLab

Commit c59e7a0c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 433844672
parent 229e43e8
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
official/vision/beta/projects/centernet/configs/experiments/coco-centernet-hourglass-gpu.yaml
View file @ c59e7a0c
...@@ -38,11 +38,11 @@ task: ...@@ -38,11 +38,11 @@ task:
per_category_metrics: false per_category_metrics: false
weight_decay: 0.0005 weight_decay: 0.0005
gradient_clip_norm: 10.0 gradient_clip_norm: 10.0
annotation_file: 'coco/instances_val2017.json' annotation_file: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/instances_val2017.json'
init_checkpoint: '/placer/prod/scratch/home/tf-model-garden-dev/vision/centernet/extremenet_hg104_512x512_coco17/2021-10-19' init_checkpoint: gs://tf_model_garden/vision/centernet/extremenet_hg104_512x512_coco17
init_checkpoint_modules: 'backbone' init_checkpoint_modules: 'backbone'
train_data: train_data:
input_path: 'coco/train*' input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*'
drop_remainder: true drop_remainder: true
dtype: 'float16' dtype: 'float16'
global_batch_size: 64 global_batch_size: 64
...@@ -57,7 +57,7 @@ task: ...@@ -57,7 +57,7 @@ task:
aug_rand_contrast: true aug_rand_contrast: true
odapi_augmentation: true odapi_augmentation: true
validation_data: validation_data:
input_path: 'coco/val*' input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*'
drop_remainder: false drop_remainder: false
dtype: 'float16' dtype: 'float16'
global_batch_size: 16 global_batch_size: 16
......
official/vision/beta/projects/centernet/configs/experiments/coco-centernet-hourglass-tpu.yaml
View file @ c59e7a0c
...@@ -37,11 +37,11 @@ task: ...@@ -37,11 +37,11 @@ task:
per_category_metrics: false per_category_metrics: false
weight_decay: 0.0005 weight_decay: 0.0005
gradient_clip_norm: 10.0 gradient_clip_norm: 10.0
annotation_file: 'coco/instances_val2017.json' annotation_file: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/instances_val2017.json'
init_checkpoint: '/placer/prod/scratch/home/tf-model-garden-dev/vision/centernet/extremenet_hg104_512x512_coco17/2021-10-19' init_checkpoint: gs://tf_model_garden/vision/centernet/extremenet_hg104_512x512_coco17
init_checkpoint_modules: 'backbone' init_checkpoint_modules: 'backbone'
train_data: train_data:
input_path: 'coco/train*' input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*'
drop_remainder: true drop_remainder: true
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 128 global_batch_size: 128
...@@ -56,7 +56,7 @@ task: ...@@ -56,7 +56,7 @@ task:
aug_rand_contrast: true aug_rand_contrast: true
odapi_augmentation: true odapi_augmentation: true
validation_data: validation_data:
input_path: 'coco/val*' input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*'
drop_remainder: false drop_remainder: false
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 64 global_batch_size: 64
......
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