Commit 002ec22b authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 393865977
parent fb603b31
...@@ -50,6 +50,7 @@ class DataConfig(cfg.DataConfig): ...@@ -50,6 +50,7 @@ class DataConfig(cfg.DataConfig):
aug_scale_min: float = 1.0 aug_scale_min: float = 1.0
aug_scale_max: float = 1.0 aug_scale_max: float = 1.0
aug_rand_hflip: bool = True aug_rand_hflip: bool = True
aug_policy: Optional[str] = None
drop_remainder: bool = True drop_remainder: bool = True
file_type: str = 'tfrecord' file_type: str = 'tfrecord'
decoder: Optional[common.DataDecoder] = common.DataDecoder() decoder: Optional[common.DataDecoder] = common.DataDecoder()
......
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