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
Show 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: ...@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 256
is_training: false is_training: false
drop_remainder: false
trainer: trainer:
checkpoint_interval: 462 checkpoint_interval: 462
optimizer_config: optimizer_config:
...@@ -58,4 +59,4 @@ trainer: ...@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462 steps_per_loop: 462
train_steps: 277200 train_steps: 277200
validation_interval: 462 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: ...@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 256
is_training: false is_training: false
drop_remainder: false
trainer: trainer:
checkpoint_interval: 462 checkpoint_interval: 462
optimizer_config: optimizer_config:
...@@ -58,4 +59,4 @@ trainer: ...@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462 steps_per_loop: 462
train_steps: 277200 train_steps: 277200
validation_interval: 462 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 # --experiment_type=retinanet_mobile_coco
# COCO mAP: 16.8 # COCO mAP: 16.5
runtime: runtime:
distribution_strategy: 'tpu' distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16' mixed_precision_dtype: 'bfloat16'
...@@ -41,8 +41,9 @@ task: ...@@ -41,8 +41,9 @@ task:
aug_scale_min: 0.5 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 256
is_training: false is_training: false
drop_remainder: false
trainer: trainer:
checkpoint_interval: 462 checkpoint_interval: 462
optimizer_config: optimizer_config:
...@@ -58,4 +59,4 @@ trainer: ...@@ -58,4 +59,4 @@ trainer:
steps_per_loop: 462 steps_per_loop: 462
train_steps: 277200 train_steps: 277200
validation_interval: 462 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