Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 336967249
parent f041db19
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/configs/maskrcnn.py
View file @ 2b2d4820
...@@ -208,7 +208,7 @@ class MaskRCNNTask(cfg.TaskConfig): ...@@ -208,7 +208,7 @@ class MaskRCNNTask(cfg.TaskConfig):
init_checkpoint_modules: str = 'all' # all or backbone init_checkpoint_modules: str = 'all' # all or backbone
annotation_file: Optional[str] = None annotation_file: Optional[str] = None
gradient_clip_norm: float = 0.0 gradient_clip_norm: float = 0.0
per_category_metrics = False per_category_metrics: bool = False
COCO_INPUT_PATH_BASE = 'coco' COCO_INPUT_PATH_BASE = 'coco'
......
official/vision/beta/configs/retinanet.py
View file @ 2b2d4820
...@@ -129,7 +129,7 @@ class RetinaNetTask(cfg.TaskConfig): ...@@ -129,7 +129,7 @@ class RetinaNetTask(cfg.TaskConfig):
init_checkpoint_modules: str = 'all' # all or backbone init_checkpoint_modules: str = 'all' # all or backbone
annotation_file: Optional[str] = None annotation_file: Optional[str] = None
gradient_clip_norm: float = 0.0 gradient_clip_norm: float = 0.0
per_category_metrics = False per_category_metrics: bool = False
@exp_factory.register_config_factory('retinanet') @exp_factory.register_config_factory('retinanet')
......
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