Commit 679b7acf authored by Rajagopal Ananthanarayanan's avatar Rajagopal Ananthanarayanan Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 350663018
parent 5067ae5d
...@@ -51,7 +51,7 @@ class DataConfig(cfg.DataConfig): ...@@ -51,7 +51,7 @@ class DataConfig(cfg.DataConfig):
aug_scale_max: float = 1.0 aug_scale_max: float = 1.0
aug_rand_hflip: bool = True aug_rand_hflip: bool = True
drop_remainder: bool = True drop_remainder: bool = True
file_type: str = 'tfrecod' # tfrecord, or sstable file_type: str = 'tfrecord' # tfrecord, or sstable
@dataclasses.dataclass @dataclasses.dataclass
......
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