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

Internal change

PiperOrigin-RevId: 433844672
parent 229e43e8
......@@ -38,11 +38,11 @@ task:
per_category_metrics: false
weight_decay: 0.0005
gradient_clip_norm: 10.0
annotation_file: 'coco/instances_val2017.json'
init_checkpoint: '/placer/prod/scratch/home/tf-model-garden-dev/vision/centernet/extremenet_hg104_512x512_coco17/2021-10-19'
annotation_file: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/instances_val2017.json'
init_checkpoint: gs://tf_model_garden/vision/centernet/extremenet_hg104_512x512_coco17
init_checkpoint_modules: 'backbone'
train_data:
input_path: 'coco/train*'
input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*'
drop_remainder: true
dtype: 'float16'
global_batch_size: 64
......@@ -57,7 +57,7 @@ task:
aug_rand_contrast: true
odapi_augmentation: true
validation_data:
input_path: 'coco/val*'
input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*'
drop_remainder: false
dtype: 'float16'
global_batch_size: 16
......
......@@ -37,11 +37,11 @@ task:
per_category_metrics: false
weight_decay: 0.0005
gradient_clip_norm: 10.0
annotation_file: 'coco/instances_val2017.json'
init_checkpoint: '/placer/prod/scratch/home/tf-model-garden-dev/vision/centernet/extremenet_hg104_512x512_coco17/2021-10-19'
annotation_file: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/instances_val2017.json'
init_checkpoint: gs://tf_model_garden/vision/centernet/extremenet_hg104_512x512_coco17
init_checkpoint_modules: 'backbone'
train_data:
input_path: 'coco/train*'
input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/train*'
drop_remainder: true
dtype: 'bfloat16'
global_batch_size: 128
......@@ -56,7 +56,7 @@ task:
aug_rand_contrast: true
odapi_augmentation: true
validation_data:
input_path: 'coco/val*'
input_path: '/readahead/200M/placer/prod/home/tensorflow-performance-data/datasets/coco/val*'
drop_remainder: false
dtype: 'bfloat16'
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