Skip to content

GitLab

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

Merge pull request #9608 from srihari-humbarwadi:patch-3 

PiperOrigin-RevId: 350881115
parents 18db08f1 3d0ac194
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/vision/beta/configs/retinanet.py
View file @ 55e8cbbf
......@@ -143,7 +143,7 @@ def retinanet() -> cfg.ExperimentConfig:
COCO_INPUT_PATH_BASE = 'coco'
COCO_TRIAN_EXAMPLES = 118287
COCO_TRAIN_EXAMPLES = 118287
COCO_VAL_EXAMPLES = 5000
......@@ -152,7 +152,7 @@ def retinanet_resnetfpn_coco() -> cfg.ExperimentConfig:
"""COCO object detection with RetinaNet."""
train_batch_size = 256
eval_batch_size = 8
steps_per_epoch = COCO_TRIAN_EXAMPLES // train_batch_size
steps_per_epoch = COCO_TRAIN_EXAMPLES // train_batch_size
config = cfg.ExperimentConfig(
runtime=cfg.RuntimeConfig(mixed_precision_dtype='bfloat16'),
......@@ -225,7 +225,7 @@ def retinanet_spinenet_coco() -> cfg.ExperimentConfig:
"""COCO object detection with RetinaNet using SpineNet backbone."""
train_batch_size = 256
eval_batch_size = 8
steps_per_epoch = COCO_TRIAN_EXAMPLES // train_batch_size
steps_per_epoch = COCO_TRAIN_EXAMPLES // train_batch_size
input_size = 640
config = cfg.ExperimentConfig(
......
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