Internal change

PiperOrigin-RevId: 391390019
SemaSandbox · Aug 17, 2021 · 555cb70 · 555cb70
1 parent 81ad46b
commit 555cb70
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/official/vision/beta/configs/retinanet.py b/official/vision/beta/configs/retinanet.py
@@ -55,6 +55,7 @@ class Parser(hyperparams.Config):
   aug_rand_hflip: bool = False
   aug_scale_min: float = 1.0
   aug_scale_max: float = 1.0
+  aug_policy: Optional[str] = None
   skip_crowd_during_training: bool = True
   max_num_instances: int = 100