Internal change

PiperOrigin-RevId: 407467989

Internal change
PiperOrigin-RevId: 407467989
01606417 · A. Unique TensorFlower · 2db7ab2a · 01606417
Commit 01606417 authored Nov 03, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/projects/edgetpu/vision/configs/semantic_segmentation_searched_config.py ...u/vision/configs/semantic_segmentation_searched_config.py +1 -1

No files found.
--- a/official/projects/edgetpu/vision/configs/semantic_segmentation_searched_config.py
+++ b/official/projects/edgetpu/vision/configs/semantic_segmentation_searched_config.py
@@ -58,7 +58,7 @@ class BiFPNHeadConfig(hyperparams.Config):


 @dataclasses.dataclass
-class Losses(hyperparams.Config):
+class Losses(base_cfg.Losses):
  label_smoothing: float = 0.0
  ignore_label: int = 255
  class_weights: List[float] = dataclasses.field(default_factory=list)