Internal change

PiperOrigin-RevId: 447027442

Internal change
PiperOrigin-RevId: 447027442
fd015075 · Abdullah Rashwan · A. Unique TensorFlower · d8d53ba8 · fd015075
Commit fd015075 authored May 06, 2022 by Abdullah Rashwan Committed by A. Unique TensorFlower May 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py ...sion/beta/projects/yolo/modeling/decoders/yolo_decoder.py +1 -1

No files found.
--- a/official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py
+++ b/official/vision/beta/projects/yolo/modeling/decoders/yolo_decoder.py
@@ -613,7 +613,7 @@ def build_yolo_decoder(
                     '{yolo_model.YOLO_MODELS[decoder_cfg.version].keys()}'
                     '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()
  for key in base_model: