Skip to content

GitLab

Commit fd015075 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 447027442
parent d8d53ba8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py
View file @ fd015075
...@@ -613,7 +613,7 @@ def build_yolo_decoder( ...@@ -613,7 +613,7 @@ def build_yolo_decoder(
'{yolo_model.YOLO_MODELS[decoder_cfg.version].keys()}' '{yolo_model.YOLO_MODELS[decoder_cfg.version].keys()}'
'or specify a custom decoder config using YoloDecoder.') 'or specify a custom decoder config using YoloDecoder.')
base_model = YOLO_MODELS[decoder_cfg.version][decoder_cfg.type] base_model = YOLO_MODELS[decoder_cfg.version][decoder_cfg.type].copy()
cfg_dict = decoder_cfg.as_dict() cfg_dict = decoder_cfg.as_dict()
for key in base_model: for key in base_model:
......
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