Skip to content

GitLab

Unverified Commit 3d0ac194 authored by Srihari Humbarwadi's avatar Srihari Humbarwadi Committed by GitHub
Browse files

fixed type in retinanet config

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