Skip to content

GitLab

Commit d3bd8c0d authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 469600523
parent ee4ab0e1
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 7 deletions
official/vision/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml
View file @ d3bd8c0d
......@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5
validation_data:
dtype: 'bfloat16'
global_batch_size: 8
global_batch_size: 256
is_training: false
drop_remainder: false
trainer:
checkpoint_interval: 462
optimizer_config:
......@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462
train_steps: 277200
validation_interval: 462
validation_steps: 625
validation_steps: 20
official/vision/configs/experiments/retinanet/coco_spinenet49s_mobile_tpu.yaml
View file @ d3bd8c0d
......@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5
validation_data:
dtype: 'bfloat16'
global_batch_size: 8
global_batch_size: 256
is_training: false
drop_remainder: false
trainer:
checkpoint_interval: 462
optimizer_config:
......@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462
train_steps: 277200
validation_interval: 462
validation_steps: 625
validation_steps: 20
official/vision/configs/experiments/retinanet/coco_spinenet49xs_mobile_tpu.yaml
View file @ d3bd8c0d
# --experiment_type=retinanet_mobile_coco
# COCO mAP: 16.8
# COCO mAP: 16.5
runtime:
distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16'
......@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5
validation_data:
dtype: 'bfloat16'
global_batch_size: 8
global_batch_size: 256
is_training: false
drop_remainder: false
trainer:
checkpoint_interval: 462
optimizer_config:
......@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462
train_steps: 277200
validation_interval: 462
validation_steps: 625
validation_steps: 20
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